知识经验分享——YOLOv5-6.0训练出错及解决方法(RuntimeError)
一、问题bug不知道小伙伴们在训练YOLOv5-6.0或者其他版本有没有遇到以下问题:
RuntimeError: result type Float can't be cast to the desired output type long int
二、解决方法
步骤1:打开utils/loss.py文件
步骤2:找到 for i in range(self.nl) 函数(Ctrl+F),作以下替换:
替换代码:
anchors, shape = self.anchors[i], p[i].shape
替换之后:
步骤3:找到# Append部分(Ctrl+F),作以下替换:
替换代码:
indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))
替换之后:
步骤4:顺利运行train.py文件
原文地址:知识经验分享——YOLOv5-6.0训练出错及解决方法(RuntimeError) - 知乎标签:YOLOv5,RuntimeError,shape,出错,6.0,替换 From: https://www.cnblogs.com/wanggang123/p/18498875