tensorRT量化yolov5模型

最新推荐文章于 2024-09-05 11:01:33 发布

「已注销」

最新推荐文章于 2024-09-05 11:01:33 发布

阅读量1.1k

点赞数 1

分类专栏：学习总结

本文链接：https://blog.csdn.net/Sept_Oct/article/details/115506462

版权

本文介绍了如何使用TensorRT加速YOLOv5模型，通过将PyTorch模型转换为TensorRT engine，实现了3-5毫秒的快速图片推断时间，不包括数据预处理和后处理步骤。

摘要由CSDN通过智能技术生成

tensorRT加速yolov5

1.下载两个项目

git clone https://github.com/wang-xinyu/tensorrtx.git
git clone https://github.com/ultralytics/yolov5.git

2.pt转wts,wts转engine

cd yolov5
sh weights/download_weights.sh
copy tensorrtx/yolov5/gen_wts.py into ultralytics/yolov5
cd ultralytics/yolov5
# 修改一些路径
python gen_wts.py
cp yolov5s.wts tensorrtx/yolov5
cd tensorrtx/yolov5
vim CMakeList.txt
# 将下面两行注释掉，这两个指向的是系统环境下tensorrt的那几个头文件的位置。
# tensorrt
# include_directories(/usr/include/x86_64-linux-gnu/)