History

wang-xinyu 4070131934 update readme		2021-04-02 09:44:28 +08:00
..
calibrator.cpp	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
calibrator.h	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
CMakeLists.txt	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
configure.h	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
gen_wts_refinedet.py	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
logging.h	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
README.md	update readme	2021-04-02 09:44:28 +08:00
refinedet.cpp	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +08:00
utils.h	tensorRT RefineDet (#443 )	2021-03-17 11:19:22 +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