Giters

DonggeunYu / tensorrt8.2_DCNv2_plugin

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TensorRT8-DCNv2-Plugin

基于TensorRT8实现DCNv2插件

prerequirements

  1. TensorRT-8.2GA

dependencies

  1. TensorRT 8.2GA

  2. onnx-tensorrt corresponding to RT8.2

  3. clone TensorRT release/8.2版本

  4. 将DCNv2文件夹、InferPlugin.cpp以及CMakeLists拷贝到TensorRT/plugin中

  5. 在TensorRT目录创建build目录，进入后运行

mkdir build && cd build
cmake .. -DBUILD_PARSERS=OFF -DBUILD_SAMPLES=OFF -DTRT_LIB_DIR=$TENSORRT_LIB_PATH -DTRT_BIN_DIR=`pwd`/out -DBUILD_PLUGINS=ON -DCUB_ROOT_DIR=$CUB_PATH
make -j4
  1. 编译完成后会生成libnvinfer_plugin库
  2. 将builtin_op_importer.cpp拷贝到onnx-tensorrt中编译libnvonnxparser库

DCNv2的实现代码摘自CaoWGG/TensorRT-CenterNet

About

License:Apache License 2.0

Languages

Language:C++ 88.2%Language:Python 4.8%Language:PureBasic 3.4%Language:Jupyter Notebook 2.3%Language:Cuda 0.9%Language:C 0.3%Language:CMake 0.1%Language:Dockerfile 0.0%Language:Shell 0.0%Language:Makefile 0.0%Language:Batchfile 0.0%Language:HTML 0.0%Language:PowerShell 0.0%