训练出错

[zhwis]

Search before asking

• I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

你好，我目前训练2类，每次训练到 epoch=59 都会出现以下错误，train 0,1,2 有向box是正确的：
File "train.py", line 631, in
main(opt)
File "train.py", line 528, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 360, in train
results, maps, _ = val.run(data_dict,
File "/opt/conda/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "/mnt/data/project/APA/yolov5_rotaion/val.py", line 237, in run
callbacks.run('on_val_image_end', pred, predn, path, names, im[si])
File "/mnt/data/project/APA/yolov5_rotaion/utils/callbacks.py", line 76, in run
logger['callback'](*args, **kwargs)
File "/mnt/data/project/APA/yolov5_rotaion/utils/loggers/init.py", line 109, in on_val_image_end
self.wandb.val_one_image(pred, predn, path, names, im)
File "/mnt/data/project/APA/yolov5_rotaion/utils/loggers/wandb/wandb_utils.py", line 485, in val_one_image
box_data = [{"position": {"minX": xyxy[0], "minY": xyxy[1], "maxX": xyxy[2], "maxY": xyxy[3]},
File "/mnt/data/project/APA/yolov5_rotaion/utils/loggers/wandb/wandb_utils.py", line 487, in
"box_caption": f"{names[cls]} {conf:.3f}",
KeyError: 2.0
关闭wandb 又会出现 WandbLogger' object has no attribute 'bbox_interval'；知道什么原因吗？

Additional

No response

[acai66]

检查下数据集配置文件里指定的nc与names的数量是否一致

[zhwis]

一致的，解决了，是pre的最后一维是angle，box_data 少解析了。
还有个问题，现在预测基本每个目标上有一个水平box，加一个旋转box，中心点及wh应该是一致的，调nms也只能抑制掉旋转box，会是角度训练还不够吗？但是loss已经收敛的很小了

[acai66]

一致的，解决了，是pre的最后一维是angle，box_data 少解析了。 还有个问题，现在预测基本每个目标上有一个水平box，加一个旋转box，中心点及wh应该是一致的，调nms也只能抑制掉旋转box，会是角度训练还不够吗？但是loss已经收敛的很小了

有可能是 角度训练未收敛，调大一点角度loss的权重试试，

yolov5_rotation/data/hyps/hyp.scratch.yaml

Line 18 in 9e38ec3

angle: 0.03

yolov5_rotation/data/hyps/hyp.scratch.yaml

Line 19 in 9e38ec3

angle_pw: 0.05 # angle BCELoss positive_weight

[Yanhui-He]

@acai66

请问如果如果我的label里的角度是浮点数，会不会影响检测的角度范围呢？
代码里没有看到对label里的角度取int整型。

[acai66]

@acai66

请问如果如果我的label里的角度是浮点数，会不会影响检测的角度范围呢？ 代码里没有看到对label里的角度取int整型。

浮点数也可以训练吗，我还没测试过，有一处取整用来生成训练目标：

yolov5_rotation/utils/loss.py

Line 109 in 9e38ec3

return np.concatenate([y_sig[math.ceil(num_class / 2) - int(label.item()):],

[Yanhui-He]

浮点数可以训练，ok，感谢，我看到了这个地方，想试一试增加精度

[Linranran]

一致的，解决了，是pre的最后一维是angle，box_data 少解析了。 还有个问题，现在预测基本每个目标上有一个水平box，加一个旋转box，中心点及wh应该是一致的，调nms也只能抑制掉旋转box，会是角度训练还不够吗？但是loss已经收敛的很小了

+1