当前位置:网站首页>运行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))
可以直接将上面代码复制上去即可。然后该错误就进行了解决。
边栏推荐
猜你喜欢
企业经营12法的领悟
麒麟信安携异构融合云金融信创解决方案亮相第十五届湖南地区金融科技交流会
【TPM2.0原理及应用指南】 16、17、18章
Functions and usage of viewflipper
Biped robot controlled by Arduino
Nerf: the ultimate replacement for deepfake?
本周小贴士#136:无序容器
With the latest Alibaba P7 technology system, mom doesn't have to worry about me looking for a job anymore
策略模式 - Unity
【TPM2.0原理及应用指南】 1-3章
随机推荐
Functions and usage of ratingbar
serachview的功能和用法
本周小贴士#136:无序容器
ViewSwitcher的功能和用法
Flash build API service
Problems encountered in Jenkins' release of H5 developed by uniapp
VSCode关于C语言的3个配置文件
Functions and usage of viewswitch
From Devops to mlops: how do it tools evolve to AI tools?
The server is completely broken and cannot be repaired. How to use backup to restore it into a virtual machine without damage?
Repair method of firewall system crash and file loss, material cost 0 yuan
LeetCode 890(C#)
How to mount the original data disk without damage after the reinstallation of proxmox ve?
数值 - number(Lua)
Enum + Validation 的个人最佳实践 demo 分享
L1-019 谁先倒(Lua)
Examen des lois et règlements sur la sécurité de l'information
状态模式 - Unity(有限状态机)
Sator推出Web3遊戲“Satorspace” ,並上線Huobi
第2章搭建CRM项目开发环境(搭建开发环境)