深度学习 计算机视觉 目标检测 【yolov5 defect报错】AttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘

项目场景：

提示：这里简述项目相关背景：

在进行利用pytorch进行yolov5目标检测训练的时候报错。

问题描述1：

yolov5报错代码如下： AttributeError: ‘Hardswish’ object has no attribute ‘inplace’

原因分析1：

主要是activation.py报错，定位到这一行代码为： def forward(self, input: Tensor) -> Tensor: return F.hardswish(input, self.inplace)

解决方案1：

原来代码为： def forward(self, input: Tensor) -> Tensor: return F.hardswish(input, self.inplace) 将代码修改为： def forward(self, input: Tensor) -> Tensor: return F.hardswish(input) 将self.inplace给删除掉

当我在尝试进行yolov5进行训练时，运行defect.py，又开始报了一种错误，具体报错内容如下：

AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

主要是upsamlping出现了问题，应该是上采样函数报错，定位到具体代码位置为： 原来代码为：

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)

再次运行defect.py就可以预测了。

相关文章

 您阅读本篇文章共花了：