0、tensorrtx官网

https://github.com/wang-xinyu/tensorrtxhttps://github.com/wang-xinyu/tensorrtx/tree/master/yolov5

1、下载源码

git clone -b v7.0 https://github.com/ultralytics/yolov5.gitgit clone -b yolov5-v7.0 https://github.com/wang-xinyu/tensorrtx.git

2、版本对应的问题

2.1 yolov5训练和预测的版本

使用新版本的yolov5训练生成的pt文件在旧版本的yolov5中预测时会报错

AttributeError: Can't get attribute 'DetectionModel' on 

解决方法:
1)升级预测时使用的yolov5版本,使之和训练时的一致;
2)参考这篇博客修改旧版本的yolov5的源码:https://blog.csdn.net/weixin_46389664/article/details/129306692

2.2 yolov5和tensorrtx的版本也要对应

参加官方说明:

https://github.com/wang-xiny