注意自己的tensorrt版本,tensortt7和tensorrt8版本的项目代码不同,混用会报错。这里仅限tensorrt7版本的yolo-tensorrt。
tensorrt7和tensorrt8版本的yolo-tensorrt项目代码,及测试用到的yolov4-tiny.weights,均放在百度云盘中,请自取。
链接:https://pan.baidu.com/s/14kg0kwfikr93UuvyHs9xLw
提取码:zg4e
1、下载测试代码
在百度云盘中获取。
cd yolo-tensorrt/configs
2、下载weights
在百度云盘中获取yolov4-tiny.weights。
将weights复制进yolo-tensorrt/configs中,如图
3、修改代码
进入/yolo-tensorrt/samples文件夹中,修改sample_detector.cpp中的两处代码(用sudo vim)
- 大概40行,将推理精度降低为int8,
config_v4_tiny.inference_precison = FP16;
- 大概51行,将模型换为yolov4-tiny,
detector->init(config_v4_tiny);
如图
4、编译运行
cd yolo-tensorrt/
sudo mkdir build
cd build/
sudo cmake -D CMAKE_CUDA_COMPILER="/usr/local/cuda/bin/nvcc" ..
sudo make
sudo ./yolo-trt