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