yolov5报错:AttributeError: No se puede obtener el atributo 'SPPF' en <módulo 'models.common' de 'E:\\All_in\\

Solución: según el mensaje de error, busque common.py

AttributeError: No se puede obtener el atributo 'SPPF' en <módulo 'models.common' de 'E:\\All_in\\新建文件夹\\yolo_double\\yolov5_stereo_Pro\\models\\common.py'>

Agregue la clase SPPF encima de spp:

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)

Luego las advertencias se marcarán en rojo, seleccione y haga clic para importar 'advertencias'

Continuar informando un error:

RuntimeError: The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3

solución:

Descarga:
https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
para reemplazar el yolov5s.pt descargado predeterminado, porque la descarga predeterminada es
el reemplazo de V6.1, después de ejecutar detect. p, está bien

Supongo que te gusta

Origin blog.csdn.net/qq_53545309/article/details/130046717
Recomendado
Clasificación