yolov5学习遇到的困难

本文观看目标检测 YOLOv5 开源代码项目调试与讲解实战【土堆 x 布尔艺数】_哔哩哔哩_bilibili视频来部署yolov5环境。

问题1：pycocotools工具包安装问题，参照视频2解决Windows平台下pycocotools错误（目标检测 YOLOv5 开源代码项目调试与讲解实战【土堆 x 布尔艺数】_哔哩哔哩_bilibili）

问题2：下载失败yolov5s.pt文件，因为github为外网可能无法访问可能导致下载失败，可自行去 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5s.pt下载（如果下载5.0版本就不会报问题3、4错误，此链接为作者提供链接）放在yolov5-5.0文件夹下即可

问题3：Can't get attribute 'SPPF' on <module 'models.common' from 'D:\\code\\yolov5-5.0\\models\\common.py'>

解决：由于视频使用的5.0版本权重已经更新到6.0版本了（个人理解）所以报错，去6.0版本里面的model/common.py里面找到SPPF的类（代码如下）把它拷到自己这个5.0版本的model/common.py里面,这样自己的代码就有这个类了,还要导入一个warnings（高亮部分）包就可以了

 1 class SPPF(nn.Module):

 2     # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher

 3     def __init__(self, c1, c2, k=5):  # equivalent to SPP(k=(5, 9, 13))

 4         super().__init__()

 5         c_ = c1 // 2  # hidden channels

 6         self.cv1 = Conv(c1, c_, 1, 1)

 7         self.cv2 = Conv(c_ * 4, c2, 1, 1)

 8         self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2)

 9

10     def forward(self, x):

11         x = self.cv1(x)

12         with warnings.catch_warnings():

13             warnings.simplefilter('ignore')  # suppress torch 1.9.0 max_pool2d() warning

14             y1 = self.m(x)

15             y2 = self.m(y1)

16             return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))

问题4：The size of tensor a (60) must match the size of tensor b (56) at non-singLeton dimension 3

解决：这是因为5.0的版本权重下载了个6.1的版本权重，所以不匹配，
可去https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt下载，放在yolov5-5.0文件夹下即可

巴特西

yolov5学习遇到的困难

最新文章

热门文章