yolov5 训练改Focal loss 一些问题

Question

yolov5 训练改Focal loss 一些问题

yolov5 训练改Focal loss 产生了几个问题，这是什么情况

1、收敛速度变慢，训练速度变慢，比初始loss慢4倍左右（正常需要1.6h、100epoch，加了之后需要7h、250epoch）
2、训练过程经常会出现nms超时警告
3、对长尾数据集map不升反降，大概降低3-4%

请问大家这些是什么原因

Answer 1

看下你的loss是怎么修改的，变慢这回事是很正常的事情，特别是学习率，或者loss的计算方式修改，都会导致你这里面的问题。
至于nms超时，一般是会在前几个或者十几个epoch出现，表示此时模型找到的结果框很多（loss修改差的情况下，权重优化不好也会这样），计算超时。一般正常稳定下来之后，预测框减少就不会了。或者你可以修改nms的耗时限制，多加一些时间，这都可以解决这个警告。

。
最后map下降，说明你的loss修改效果不如旧版本啊，需要再次修改成其他方案。
目前来说，除非是特定的数据集，不然一般默认的loss是很多人在很多数据集上面优化多次的结果（特别是类似这种星星多的项目），所以你的修改还真的不一定比原始的好。