解决YOLO v5的AttributeError: Can‘t get attribute SPPF on module models. common的问题
·
**
一.问题描述
在运行YOLOv5的detect.py程序时候,出现了以下错误提示AttributeError: Can‘t get attribute SPPF on module models. common from D://yolov\yolov5-5.0\models\common.py**
二.解决方法
- 在GitHub中下载YOLOv6,打开文件找到models文件夹:
- 在models文件夹中找到common.py,打开后使用查找SPPF关键词,找到文件中SPPF类,将下列代码与之替换:
class SPPF(nn.Module): # export-friendly version of nn.SiLU()
@staticmethod
def forward(x):
return x * torch.sigmoid(x)
问题解决:
三.参考来源
更多推荐
已为社区贡献2条内容
所有评论(0)