Yolov5各种bug记录

Yolov5各种bug记录,第1张

运行yolov5这个经典github项目的detect.py时遇到了许多bug,这里列举出我遇到的bug以及我搜索到的解决方案(如果原作者感觉冒犯,联系我立马删掉(乖巧.jpg))

bug1:

subprocess.CalledProcessError: Command ‘git tag’ returned non-zero exit stat

解决方案:
关闭vpn,或点击查看原文链接

bug2:

AttributeError: Can’t get attribute ‘SPPF’ on

解决方案:
在models文件夹下面的common文件中添加以下代码,或点击查看原文链接

import warnings

class SPPF(nn.Module):
    # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
    def __init__(self, c1, c2, k=5):  # equivalent to SPP(k=(5, 9, 13))
        super().__init__()
        c_ = c1 // 2  # hidden channels
        self.cv1 = Conv(c1, c_, 1, 1)
        self.cv2 = Conv(c_ * 4, c2, 1, 1)
        self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2)
 
    def forward(self, x):
        x = self.cv1(x)
        with warnings.catch_warnings():
            warnings.simplefilter('ignore')  # suppress torch 1.9.0 max_pool2d() warning
            y1 = self.m(x)
            y2 = self.m(y1)
            return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))
bug3:

AssertionError: Image Not Found D:…
明明图片就在这个路径下但是却说不在这我也是很懵了

解决方案:
更改tils\datasets.py 中的 LoadImages类中的图片路径,或点击查看原文链接

# p = str(Path(path).absolute())  # os-agnostic absolute path
p = str(Path(path)) # 注释掉上面那句,换成这句
bug4:

AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

解决方案:

点击可查看原文链接

点击下图框出来的代码链接进去修改:

找到这一块代码,按照图片所示替换:

替换代码:

return F.interpolate(input,self.size,self.scale_factor,self.mode,self.align_corners)
bug5:

是的你没看错,我也没想到还有bug
RuntimeError: The size of tensor a (60) must match the size of tensor b (56) at non-singleton dimension 3

原因是跟着b站up主学代码下载的这个5.0版本下载了一个6.1的模型,导致不匹配。

解决方案:

点击查看原文链接

点击下面这个链接去重新下载一个yolov5s.pt文件:
下载yolov5s.pt

终于运行成功了,感谢互联网上的兄弟姐妹们提供的解决方案~,如有问题,欢迎批评指正!

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/918827.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-16
下一篇 2022-05-16

发表评论

登录后才能评论

评论列表(0条)

保存