History

haotian 7ac785ddb3 first commit		2025-03-07 11:35:40 +08:00
..
calibrator.cpp	first commit	2025-03-07 11:35:40 +08:00
calibrator.h	first commit	2025-03-07 11:35:40 +08:00
configure.h	first commit	2025-03-07 11:35:40 +08:00
gen_wts_refinedet.py	first commit	2025-03-07 11:35:40 +08:00
logging.h	first commit	2025-03-07 11:35:40 +08:00
README.md	first commit	2025-03-07 11:35:40 +08:00
refinedet.cpp	first commit	2025-03-07 11:35:40 +08:00
utils.h	first commit	2025-03-07 11:35:40 +08:00

README.md

RefineDet

For the Pytorch implementation, you can refer to luuuyi/RefineDet.PyTorch

How to run

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

dependence

TensorRT7.0.0.11 
OpenCV >= 3.4
libtorch >=1.1.0

feature

1.tensorrt Multi output
2.L2norm
3.Postprocessing with libtorch

More Information

See the readme in home page.
tensorrt tutorials
For more detailed guidance, see yhl blog