【基于YOLOV5模型的tensorrt部署】

VIP文章 weixin_40744387

已于 2023-02-02 19:05:22 修改

阅读量429

点赞数

文章标签： git python github

于 2023-02-02 19:05:00 首次发布

本文链接：https://blog.csdn.net/weixin_40744387/article/details/128855688

版权

基于YOLOV5模型的tensorrt部署

应用案例：loge&二维码检测

yolov5模型训练参考yolov5

yolov5模型转换tensorrt模型方法参考tensorrtx
步骤如下：

步骤一：

将tensorrtx工程下的gen_wts.py拷贝到 {ultralytics}/yolov5工程下，在 {ultralytics}/yolov5工程下将训练完的pt模型转为wts模型
cp {tensorrtx}/yolov5/gen_wts.py {ultralytics}/yolov5
cd {ultralytics}/yolov5
python gen_wts.py -w yolov5s.pt -o yolov5s.wts

碰到的问题
运行找不到utils，需要去yolov5原工程下去运行gen_wts.py，不是在tensorrtx/yolov5路径下运行

步骤二：

cd {tensorrtx}/yolov5/
mkdir build
cd build
cp {ultralytics}/yolov5/yolov5s.wts {tensorrtx}/yolov5/build
cmake ..
make
./yolov5 -s yolov5s.wts yolov5s.engine s