使用TensorRT加速ScaledYOLOv4
很多人都写过TensorRT版本的yolo了,我也来写一个。具体代码可参见我的github
测试环境
ubuntu 18.04
pytorch 1.7.1
jetpack 4.4
CUDA 11.0
TensorRT7.1
快速开始
1、生成onnx模型
git clone --branch yolov4-csp https://github.com/WongKinYiu/ScaledYOLOv4
git clone https://github.com/talebolano/TensorRT-Scaled-YOLOv4
cp TensorRT-Scaled-YOLOv4/script/ScaledYOLOv4-csp/* ScaledYOLOv4/
下载yolov4-csp.weights到ScaledYOLOv4/
cd ScaledYOLOv4
python3 export.py