当前位置:   article > 正文

YOLOx添加Focal loss

yolox添加focal loss

将YOLOx中的BCE loss更改为Focal loss
loss函数分为三部分,位置损失、置信度损失、类别损失,此处只需要将置信度损失更换为Focal loss,具体原理请仔细理解置信度损失的含义。
YOLOX链接:https://link.zhihu.com/?target=https%3A//github.com/Megvii-BaseDetection/YOLOX
直接在YOLOX-main/yolox/models/yolo_head.py的YOLOXHead类中创建focal_loss方法

1 找到置信度预测损失计算位置loss_obj,并进行替换(位置在386-405行左右)
loss_iou:定位损失;loss_obj:置信度预测损失;loss_cls:预测损失
        loss_iou = (
            self.iou_loss(bbox_preds.view(-1, 4)[fg_masks], reg_targets)
        )
  • 1
  • 2
  • 3
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/384873
推荐阅读
相关标签
  

闽ICP备14008679号