AttributeError: Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘ from ‘model\\common.py‘

最新推荐文章于 2025-11-10 19:55:16 发布

原创最新推荐文章于 2025-11-10 19:55:16 发布 · 4.9k 阅读

11 ·

CC 4.0 BY-SA版权

文章标签：

#计算机视觉 #机器学习 #深度学习 #神经网络 #目标检测

这篇博客介绍了如何在YOLOv5模型中添加Spatial Pyramid Pooling-Fast (SPPF)层，通过简化版本的时空金字塔池化来提高特征提取效率。作者提供了详细的代码实现步骤，包括类定义和forward函数的处理，以及如何忽略警告以进行模型训练。

解决方案

找到common.py文件，在里面添加如下代码：

import warnings


class SPPF(nn.Module):
    # Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
    def __init__(self, c1, c2, k=5):
        super().__init__()
        c_ = c1 // 2  # hidden channels
        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))