Tensor: return F.interpolate(input, self.size, self.scal_upsampl">
当前位置:   article > 正文

AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘_upsampling.py

upsampling.py

yolov5运行detect.py文件出现错误

提示

  File "/home/taobotics/rosenv/lib/python3.7/site-packages/torch/nn/modules/upsampling.py", line 154
  • 1

去到这个目录修改upsampling.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)
  • 1
  • 2

改为

def forward(self, input: Tensor) -> Tensor:
        return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
  • 1
  • 2

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/SSS__jq/article/details/123458804
————————————————

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/888903
推荐阅读
相关标签
  

闽ICP备14008679号