Pelee-TensorRT - 移动设备上的实时对象检测系统
Pelee(NeurIPS'18)-TensorRT Implementation
C/C++ 机器学习
共83Star
详细介绍
Pelee-TensorRT using Caffe Parser
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
-
841 Star
-
0 Star
-
224 Star
-
5657 Star
-
598 Star
-
0 Star
-
48 Star
-
3948 Star