haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
0c78095dd6
duan8/centernet
History
SsisyphusTao d9e2cbc294
add centernet dla34 ctdet task. (#558) 
* add centernet dla34 ctdet task.

* update readme.

* update readme and fix a bug in sample.

Co-authored-by: chandler <chandler@invix.com>
2021-05-23 16:29:14 +08:00
..
dcnv2Plugin add centernet dla34 ctdet task. (#558) 2021-05-23 16:29:14 +08:00
sample add centernet dla34 ctdet task. (#558) 2021-05-23 16:29:14 +08:00
centernet.py add centernet dla34 ctdet task. (#558) 2021-05-23 16:29:14 +08:00
README.md add centernet dla34 ctdet task. (#558) 2021-05-23 16:29:14 +08:00

README.md

CenterNet

This is the trt implementation of detection model ctdet_coco_dla_2x from xingyizhou/CenterNet official work.

How to Run

  1. Follow NVIDIA/TensorRT tutorial to build TensorRT7

  2. Copy folder dcnv2Plugin to TensorRT/plugin and edit InferPlugin.cpp and CMakeLists.txt

  3. Rebuild to install custom plugin

  4. Use tensorrt-7.2.3.4-cp36-none-linux_x86_64.whl in TensorRT OSS to update your python-tensorrt

  5. Run python centernet.py -m ${PTH_PATH} -s to create trt engine

Sample

// Download ctdet_coco_dla_2x.pth and transfer it into trt engine first
// Download the test img from https://raw.githubusercontent.com/tensorflow/models/master/research/deeplab/g3doc/img/image2.jpg or choose your own one
cd sample
python test.py ${ENGINE_PATH} ${IMG_PATH}

trt_out

TODO

Integrate the post process with trt engine to make it more easier to use.