YOLOv5 报错怎么解决

Posted 2022-03-14 weixin_44233647

Traceback (most recent call last):
  File "E:\\yolo\\yolov5-5.0\\yolov5-5.0\\detect.py", line 178, in <module>
    detect()
  File "E:\\yolo\\yolov5-5.0\\yolov5-5.0\\detect.py", line 72, in detect
    pred = model(img, augment=opt.augment)[0]
  File "D:\\python\\anaconda\\lib\\site-packages\\torch\\nn\\modules\\module.py", line 1110, in _call_impl
    return forward_call(*input, **kwargs)
  File "E:\\yolo\\yolov5-5.0\\yolov5-5.0\\models\\yolo.py", line 123, in forward
    return self.forward_once(x, profile)  # single-scale inference, train
  File "E:\\yolo\\yolov5-5.0\\yolov5-5.0\\models\\yolo.py", line 139, in forward_once
    x = m(x)  # run
  File "D:\\python\\anaconda\\lib\\site-packages\\torch\\nn\\modules\\module.py", line 1110, in _call_impl
    return forward_call(*input, **kwargs)
  File "D:\\python\\anaconda\\lib\\site-packages\\torch\\nn\\modules\\upsampling.py", line 154, in forward
    recompute_scale_factor=self.recompute_scale_factor)
  File "D:\\python\\anaconda\\lib\\site-packages\\torch\\nn\\modules\\module.py", line 1185, in __getattr__
    raise AttributeError("'' object has no attribute ''".format(
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'