文件名称:Pelee(NeurIPS'18)-TensorRT实现(Caffe Parser)-C/C++开发
文件大小:29.96MB
文件格式:ZIP
更新时间:2024-06-15 18:54:06
C/C++ Machine Learning
Pelee(NeurIPS'18)-TensorRT的实现使用Caffe Parser的Pelee-TensorRT通过TensorRT加速Pelee Pelee:移动设备上的实时对象检测系统(NeurIPS 2018)TensorRT-Pelee可以在Jetson TX2(FP32)上运行70FPS(11ms)以上)性能(FP32)NVIDIA Jetson TX2:72 FPS(13.2〜11 ms)Titan V:200 FPS(5 ms)要求:TensorRT 4.x(Jetpack 3.3)CUDA 9.0 cudnn 7.运行:cmake。 make ./build/bin/pelee待办事项:FP16实现将自定义层IPlugin更改为IPluginExt该错误已修复。
【文件预览】:
Pelee-TensorRT-master
----common.cpp(23KB)
----mathFunctions.cpp(2KB)
----imageBuffer.h(1KB)
----cudaUtility.h(2KB)
----cmake_install.cmake(3KB)
----model()
--------pelee()
----kernel.cu(7KB)
----common.h(890B)
----pluginImplement.h(15KB)
----testVideo()
--------test.avi(729KB)
----testPic()
--------test.png(160KB)
----tensorNet.h(2KB)
----mathFunctions.h(4KB)
----CMakeLists.txt(3KB)
----main.cpp(5KB)
----util()
--------loadImage.cpp(5KB)
--------cuda()
--------loadImage.h(3KB)
----mathFunctions.cu(1KB)
----README.md(632B)
----pluginImplement.cpp(54KB)
----.gitignore(1KB)
----tensorNet.cpp(9KB)