【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)
【yolov5 defect报错】AttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘_第1张图片


解决方案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报错】AttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘_第2张图片

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

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

【yolov5 defect报错】AttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘_第3张图片
主要是upsamlping出现了问题,应该是上采样函数报错,定位到具体代码位置为:
【yolov5 defect报错】AttributeError: ‘Hardswish‘ object has no attribute ‘inplace‘_第4张图片
原来代码为:

    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就可以预测了。

你可能感兴趣的:(深度学习,计算机视觉,目标检测)