TRTForYolov3

April 19, 2019 ยท View on GitHub

Desc

tensorRT for Yolov3

alt tag

Test Enviroments

Ubuntu  16.04
TensorRT 4.0.1.6
CUDA 9.0

Models

Download the caffe model converted by official model:

Run Sample

#build source code
git submodule update --init --recursive
mkdir build
cd build && cmake .. && make && make install
cd ..

#for mobilenet-yolov3-608

> sh demo.sh

Performance

ModelGPUModeInference Time
mobilenet-yolov3-608GTX 1080float3212.655ms
mobilenet-yolov3-608Jetson nanofloat32247.655ms
mobilenet-yolov3-lite-320Jetson nanofloat32114.055ms
mobilenet-yolov3-lite-320Jetson nanofloat1697.189 ms
  • The performance in jetson nano was not satisfactory , I will try to optimize it

Details About Wrapper

see link TensorRTWrapper