Accelerate Pelee with TensorRT Pelee: A Real-Time Object Detection System on Mobile Devices (NeurIPS 2018)
TensorRT-Pelee can run over 70FPS(11ms) on Jetson TX2(FP32)
Performance(FP32)
- NVIDIA Jetson TX2: 72 FPS (13.2~11 ms)
- Titan V: 200 FPS (5 ms)
Requirements:
- TensorRT 4.x (Jetpack 3.3)
- CUDA 9.0
- cudnn 7.
Run:
cmake .
make
./build/bin/pelee
TODO:
- FP16 Implementation
- Change Custom layers IPlugin to IPluginExt
The bug has been fixed