使用YOLOv5做目标检测的时候需要注意的地方

在源代码的loss.py中有这么一句

indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0,gain[2]-1))) 

直接训练的话汇报数据类型的错误，大概的意思是浮点数不能转换成长整型，因为indices是个只能存放整数的数组，具体的原因是在最新版的torch中clamp_方法改过了，现在要改成clamp,在加上long（）确保是整型数据：

indices.append((b, a, gj.clamp(0, gain[3] - 1).long(), gi.clamp(0,gain[2]-1).long()))