在使用yolov5-5.0训练的时候,会出现一些报错。

如 

AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

意思是:

“Upsample”对象没有属性“recompute_scale_factor”

我们根据报错的提示,进入到upsampling.py,找到 forward()函数。


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)


注释掉,改成下面这个代码。为什么注释,因为以便别的项目使用,以后要改回来。upsampling.py并不是yolov5的源码。


def forward(self, input: Tensor) -> Tensor:

    return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)


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

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐