赞
踩
将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方法
loss_iou:定位损失;loss_obj:置信度预测损失;loss_cls:预测损失
loss_iou = (
self.iou_loss(bbox_preds.view(-1, 4)[fg_masks], reg_targets)
)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。