基于 https://blog.csdn.net/qq_42362891/article/details/116107108
YOLOv5 部署的 评论区 进一步degug
tensorrtx目前支持很多种网络(vgg,googlenet,yolov5等)的转换,这里我们主要关注yolov5
注意:由于不同版本的Yolov5网络架构不同,所以这里Yolov5和tensorrtx也有版本对应
首先看一下tensorrtx的Tags,会发现他目前最高只支持到 yolov5-v5.0
而yolov5最新版本是6.1,所以我们需要退而求其次,下载yolov5-v5.0
注意:在releases中下载的,可能没有注意到 yolov5的版本,直接下载了最新的。
在继续执行 即可
https://github.com/wang-xinyu/tensorrtx/tree/yolov5-v5.0/yolov5
python gen_wts.py =2 yolov5s.pt -o yolov5s.wts