For the Pytorch implementation, you can refer to luuuyi/RefineDet.PyTorch
1. generate wts file. from pytorch
python gen_wts_refinedet.py
// a file 'refinedet.wts' will be generated.
2. build tensorrtx/RefineDet and run or Using clion to open a project(recommend)
Configuration file in configure.h
You need configure your own paths and modes(SERIALIZE or INFER)
Detailed information reference configure.h
mkdir build
cd build
cmake ..
make
TensorRT7.0.0.11
OpenCV >= 3.4
libtorch >=1.1.0
1.tensorrt Multi output
2.L2norm
3.Postprocessing with libtorch
See the readme in home page.
tensorrt tutorials
For more detailed guidance, see yhl blog