当前位置:网站首页>运行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))
可以直接将上面代码复制上去即可。然后该错误就进行了解决。
边栏推荐
- MySQL usage notes 1
- What is cloud computing?
- Siggraph 2022 best technical paper award comes out! Chen Baoquan team of Peking University was nominated for honorary nomination
- 【分布式理论】(一)分布式事务
- Linux 安装mysql8.X超详细图文教程
- From Devops to mlops: how do it tools evolve to AI tools?
- Matplotlib绘图界面设置
- 企业经营12法的领悟
- 【网络攻防原理与技术】第1章:绪论
- calendarview日历视图组件的功能和用法
猜你喜欢
Function and usage of numberpick
Lex & yacc of Pisa proxy SQL parsing
专精特新软件开发类企业实力指数发布,麒麟信安荣誉登榜
Create dialog style windows with popupwindow
【重新理解通信模型】Reactor 模式在 Redis 和 Kafka 中的应用
DatePickerDialog和trimepickerDialog
Is AI more fair than people in the distribution of wealth? Research on multiplayer game from deepmind
Numberpick的功能和用法
麒麟信安中标国网新一代调度项目!
【可信计算】第十三次课:TPM扩展授权与密钥管理
随机推荐
【信息安全法律法規】複習篇
99%的人都不知道|私有化部署还永久免费的即时通讯软件!
[source code interpretation] | source code interpretation of livelistenerbus
网络攻防复习篇
99% of users often make mistakes in power Bi cloud reports
Linux 安装mysql8.X超详细图文教程
L1-027 出租(Lua)
命令模式 - Unity
赋能智慧电力建设 | 麒麟信安高可用集群管理系统,保障用户关键业务连续性
如何在软件研发阶段落地安全实践
Siggraph 2022 best technical paper award comes out! Chen Baoquan team of Peking University was nominated for honorary nomination
LeetCode 890(C#)
【网络攻防原理与技术】第3章:网络侦察技术
专精特新软件开发类企业实力指数发布,麒麟信安荣誉登榜
[Fantan] how to design a test platform?
Functions and usage of tabhost tab
Functions and usage of viewflipper
[Huang ah code] Why do I suggest you choose go instead of PHP?
User defined view essential knowledge, Android R & D post must ask 30+ advanced interview questions
L1-023 输出GPLT(Lua)