当前位置:网站首页>运行yolo v5-5.0版本报错找不到SPPF错误,进行解决
运行yolo v5-5.0版本报错找不到SPPF错误,进行解决
2022-07-07 15:40:00 【AI炮灰】
首先找到yolov5下面的models文件夹下面的common.py文件,如下所示:
找到该文件夹下面的SPP类,
然后在该类的上面添加SPPF类,如下所示,
class SPPF(nn.Module):
def __init__(self, c1, c2, k = 5):
super().__init__()
c_ = c1 // 2
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')
y1 = self.m(x)
y2 = self.m(y1)
return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))
可以直接将上面代码复制上去即可。然后该错误就进行了解决。
边栏推荐
猜你喜欢
随机推荐
到底有多二(Lua)
Audio Device Strategy 音频设备输出、输入 选择 基于7.0 代码
【饭谈】如何设计好一款测试平台?
命令模式 - Unity
2021-06-28
99%的人都不知道|私有化部署还永久免费的即时通讯软件!
calendarview日历视图组件的功能和用法
管理VDI的几个最佳实践
【分布式理论】(一)分布式事务
【分布式理论】(二)分布式存储
【OKR目标管理】价值分析
百度地图自定义样式向右拖拽导致全球地图经度0度无法正常显示
【TPM2.0原理及应用指南】 9、10、11章
Sator a lancé le jeu web 3 "satorspace" et a lancé huobi
【重新理解通信模型】Reactor 模式在 Redis 和 Kafka 中的应用
数值 - number(Lua)
LeetCode 648(C#)
本周小贴士#140:常量:安全习语
【信息安全法律法規】複習篇
青年时代历练和职业发展