backbone | input_size | GPU | mode | inference Time |
---|---|---|---|---|
mobilenetv2 | 512x512 | gtx 1070 | float32 | 3.798ms |
mobilenetv2 | 512x512 | jetson TX2 | float16 | 22ms |
- gtx 1070
ubuntu 1604
TensorRT 5.0
- jetson TX2
jetpack 4.2
- Convert CenterNet model to onnx (deform conv is not support)
- Use netron to observe whether the output of the converted onnx model is (hm, reg, wh)
git clone https://github.com/CaoWGG/TensorRT-CenterNet.git
cd TensorRT-CenterNet
mkdir build
cd build && cmake .. && make
cd ..
##cthelmet
./buildEngine -i model/ctdet_helmet.onnx -o model/ctdet_helmet.engine
./runDet -i model/ctdet_helmet.engine -img test.jpg -cap test.h264
##centerface
./buildEngine -i model/centerface.onnx -o model/centerface.engine
./runDet -i model/centerface.engine -img test.jpg -cap test.h264