YOLO损失函数中的不平衡数据补救措施

时间:2019-06-18 16:33:43

标签: tensorflow loss-function yolo

据我了解,YOLO损失函数具有权重以补偿“ noobj”的影响,这意味着在所讨论的网格单元中没有对象,因此无需知道位置值或概率对象分类。论文https://arxiv.org/pdf/1506.02640.pdf提到,noobj的权重为0.5,存在的对象的权重为5。

我正在处理另一个数据集。我有8个对象要检测,并且我正在使用具有类似YOLO模型头的截短的Mobilenet体系结构来实现对象检测。我有大量的失衡数据,比率为0.0095。

由于0.5和5不能给我很好的结果,如何找到好的体重数字? (0.0095比1也不太好)。还有其他方法可以解决这个问题吗?

0 个答案:

没有答案