yolov5 6.1训练遇到的问题

项目场景：

yolov5 6.1训练遇到的问题

问题描述

raise AttributeError("module {!r} has no attribute "
AttributeError: module ‘numpy’ has no attribute ‘int’

原因分析：

numpy版本太高，下一个低版本即可

解决方案：

pip install numpy==1.21 -i https://pypi.tuna.tsinghua.edu.cn/simple

问题描述

indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) image, anchor, grid indicesRuntimeError: result type Float can’t be cast to the desired output type long int

解决方案：

修改loss.py
大概第173行修改

 gain = torch.ones(7, device=targets.device).long()  # normalized to gridspace gain