Yolov5进阶之八 高低版本格式转换问题

本文介绍了如何解决Yolov5 5.0版本无法直接使用6.0及更高版本权重的问题,包括SPPF类缺失、YouTube URL错误和维度不匹配等错误的解决方法,提供了详细的代码修改步骤。
摘要由CSDN通过智能技术生成

yolov5 6.0 之后 pt权重文件发生了变化,不能于5.0 pt混用,导致原有5.0版本很多时候要自己训练数据集。
下面提供一种yolov5 5.0 调用 最新 yolov5 6.0 6.1 权重文件的方法。
直接适用 6.0 数据集,执行

python detect.py --source 0

AttributeError: Can’t get attribute ‘SPPF’ on <module ‘models.common’ from ‘D:\PycharmProjects\deepsort3.0\Yolov5_StrongSORT_OSNet-3.0\yolov5\models\common.py’>
是由于 5.0 没有SPPF 类,将6.0 的SPPF类复制到model下的common.py 并在头部加入 import warnings
继续执行上面命令,提示
if ‘youtube.com/’ in url or ‘youtu.be/’ in url: # if source is YouTube video
TypeError: argument of type ‘int’ is not iterable
这个问题进阶六解释了,只需要在 dataset.py 里url变为 str(url),连续两处。
再次执行,
RuntimeError: The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3
这个最难办,就是新版本这个函数2维变3维了,打开model下的yolo.py,准备把 6.0 的yolo.py 部分赋值过来,两个函数 def forward(self, x):和def _make_grid(self, nx=20, ny=20, i&#

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

宇称不守恒4.0

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值