yolov5s-5.0下的错误：result type Float can‘t be cast to the desired output type __int64

今天在对yolov5进行训练的时候出现了以下的错误

 该错误在之前的原始版本yolov5中并没有遇到过，这次是新出现的问题，yolov5的版本是5.0。以下是记录的修改部分。

主要修改了两处代码，都是在yolov5/utils/loss.py文件里的。其中第一处修改的地方为：

loss.py里的第178行代码，修改后如下所示，其中前面的是修改后的内容，注释掉的是修改前的内容。

第二处是第211行的代码，修改后的代码在212行，211行被注释掉的部分是修改前的内容。

indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))  # image, anchor, grid

然后就可以运行了。结果如下

 

参考了博客修改的问题。http://t.csdn.cn/D7soK