Skip to content

Latest commit

 

History

History
 
 

centernet

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

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.