DCNv2 plugin implemented on TensorRT7
2021-10-27 support Tensorrt8 (Since TensorRT8 make many method noexcept,so you have to add noexcept to the method)
- CUB-1.8.0
- TensorRT-7.0.0.11
-
TensorRT 7.0
-
onnx-tensorrt 7.0
-
clone TensorRT release/7.0
-
put DCNv2 directory、InferPlugin.cpp and CMakeLists to TensorRT/plugin
cd TensorRT
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
- after compile you will have libnvinfer_plugin.so with DCNv2
- put builtin_op_importer.cpp to onnx-tensorrt and compile onnx-tensorrt to get libnvonnxparser.so
- use libnvinfer_plugin.so and libnvonnxparser.so replace the origin so in TensorRT/lib
code of DCNv2 come from CaoWGG/TensorRT-CenterNet