linux使用技巧

运行yolov5出现问题AttributeError: Cant get attribute SPPF on module models.common

问题描述：

运行yolov5下面Tags5的代码出现问题：

Traceback (most recent call last):
  File "/media/ls/娱乐/yolov5/yolov5-5.0/detect.py", line 192, in <module>
    detect()
  File "/media/ls/娱乐/yolov5/yolov5-5.0/detect.py", line 34, in detect
    model = attempt_load(weights, map_location=device)  # load FP32 model
  File "/media/ls/娱乐/yolov5/yolov5-5.0/models/experimental.py", line 118, in attempt_load
    ckpt = torch.load(w, map_location=map_location)  # load
  File "/home/ls/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 607, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/ls/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 882, in _load
    result = unpickler.load()
  File "/home/ls/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/serialization.py", line 875, in find_class
    return super().find_class(mod_name, name)
AttributeError: Can't get attribute 'SPPF' on <module 'models.common' from '/media/ls/娱乐/yolov5/yolov5-5.0/models/common.py'>

Process finished with exit code 1

遇到这个问题提之后我在百度查了很久，都没找到答案。

最后我在某个地方的评论区找到了某个大佬的回复，最终得到解决方案，特此公布给遇到同样问题的人。

我这边运行的是yolov5下面Tags5的代码,出现了这个报错：
AttributeError: Cant get attribute SPPF on module models.common from ‘/media/ls/娱乐/yolov5/yolov5-5.0/models/common.py’>

解决方案是：

去Tags6里面的model/common.py里面去找到这个SPPF的类,把它拷过来到你这个Tags5的model/common.py里面,这样你的代码就也有这个类了,还要引入一个warnings包就行了！

有的同学找不到SPPF这个类，那我现在直接粘贴在这里，你们只需要复制到你们的common.py里面即可，记得把import warnings放在上面去：

import warnings

class SPPF(nn.Module):
# Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocher
def __init__(self, c1, c2, k=5):  # equivalent to SPP(k=(5, 9, 13))
    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')  # suppress torch 1.9.0 max_pool2d() warning
        y1 = self.m(x)
        y2 = self.m(y1)
        return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))

亲测运行成功!!!