Model Summary: 270 layers, 7235389 parameters, 0 gradients
image 1/2 D:\pythonProject\ve\Lib\site-packages\yolov5\data\images\bus.jpg: Traceback (most recent call last):
File "D:\pythonProject\ve\Lib\site-packages\yolov5\detect.py", line 311, in
main()
File "D:\pythonProject\ve\Lib\site-packages\yolov5\detect.py", line 307, in main
run(**vars(opt))
File "D:\pythonProject\ve\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "D:\pythonProject\ve\Lib\site-packages\yolov5\detect.py", line 155, in run
pred = model(img, augment=augment, visualize=visualize)[0]
File "D:\pythonProject\ve\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\pythonProject\ve\lib\site-packages\yolov5\models\yolo.py", line 125, in forward
return self._forward_once(x, profile, visualize) # single-scale inference, train
File "D:\pythonProject\ve\lib\site-packages\yolov5\models\yolo.py", line 148, in _forward_once
x = m(x) # run
File "D:\pythonProject\ve\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "D:\pythonProject\ve\lib\site-packages\torch\nn\modules\upsampling.py", line 154, in forward
recompute_scale_factor=self.recompute_scale_factor)
File "D:\pythonProject\ve\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'
Process finished with exit code 1
解决了
把upampling.py
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
recompute_scale_factor=self.recompute_scale_factor)
改成
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
就好了
升级pytorch至1.11.0看看可否解决问题,参考:
https://github.com/ultralytics/yolov5/issues/6948
你好,问题解决了吗?我也有同样的问题,请问怎么解决呀
recompute_scale_factor=self.recompute_scale_factor)
self对象中没有.recompute_scale_factor属性
可能是安装的yolov5 模块版本不对