bubbliiiing / yolox-pytorch

这是一个yolox-pytorch的源码,可以用于训练自己的模型。
Apache License 2.0
877 stars 184 forks source link

损失函数替换为Focal Loss #127

Open SCcdet opened 1 year ago

SCcdet commented 1 year ago

B导,你这个能将loss_obj替换为Focal Loss么,为什么我尝试替换后loss_obj在训练过程中一直为0啊?

bubbliiiing commented 1 year ago

已经做了正负样本的平衡,没啥必要Focal,效果不太好

SCcdet commented 1 year ago

好的,谢谢B导