训练出错

[devil1213]

Search before asking

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

Question

大佬您好，我在用您的代码训练有向框时 训练结束出现了
Class Images Labels P R [email protected] mAP@WARNING: NMS time limit 10.0s exceeded
Class Images Labels P R [email protected] mAP@WARNING: NMS time limit 10.0s exceeded
一直循环这个 请问怎么解决 然后在结果里面也没有map的结果

Additional

No response

[acai66]

旋转框nms效率低造成超时，模型收敛后nms速度还可以，训练前期可以先不评估，后期再开始评估，有个训练参数用来设置什么时候开始评估，调大点试试

--begin_val默认为第30epoch。

[devil1213]

好的 谢谢大佬 我改了主干网络 好像运行不出来了 之前在水平框可以 好像改的的模块也要转换成有向框的格式吗

[acai66]

好的 谢谢大佬 我改了主干网络 好像运行不出来了 之前在水平框可以 好像改的的模块也要转换成有向框的格式吗

改主干网络是不影响的，只有Detect类

yolov5_rotation/models/yolo.py

Line 33 in 9e38ec3

class Detect(nn.Module):

有影响

[devil1213]

好的大佬 我改好了 正在训练 就是速度太慢了哈哈哈哈 一个晚上就训练了26epoch，，，，