项目场景:
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