false positives 特别多的问题

yatengLG / Retinanet-Pytorch

Retinanet目标检测算法(简单,明了,易用,全中文注释,单机多卡训练,视频检测)(based on pytorch,Simple, Clear, Mutil GPU)

267 stars 96 forks source link

Open xzxedu opened 3 years ago

xzxedu commented 3 years ago

您好，非常感谢您的代码。我尝试将您的focal loss部分加到retinaface模型中，没有改动参数，发现误检特别多。想问下知道可能的原因嘛？是因为我没有进行hard negative mining还是因为我参数没有调对呢？(尝试过修改阈值但作用非常微小)

yatengLG commented 3 years ago

我也不太清楚是什么原因导致您这边误检率特别高。

我只能给您说下，这边为什么这么做。α是一个调节正负例的参数，是负例相对于整体的比重；γ是调节识别难易度的参数，一般2就可以了，不需要动。 https://github.com/yatengLG/Retinanet-Pytorch/blob/bcf56192ad8fa668100593b6b37b33a550fe8bf6/Model/struct/Focal_Loss.py#L8

xzxedu commented 3 years ago

好的谢谢～