SlideLoss,EMASlideLoss.(可动态调节正负样本的系数,让模型更加注重难分类,错误分类的样本上)Yolo-Face V2
在ultralytics/utils/loss.py中的class v8DetectionLoss进行设定.(支持v8-detect、v8-seg、v8-pose、v10) EMASlideLoss具体思想可以参考:https://www.bilibili.com/video/BV1W14y1i79U/?vd_source=c8452371e7ca510979593165c8d7ac27
FocalLoss,VarifocalLoss,QualityfocalLoss(支持v8-detect、v8-seg、v8-pose、v8-obb、v10)
项目视频百度云链接-20240111版本更新说明.
Normalized Gaussian Wasserstein Distance(支持v8-detect、v8-seg、v8-pose、v10)论文链接
在Loss中使用:
在ultralytics/utils/loss.py中的BboxLoss class中的__init__函数里面设置self.nwd_loss为True.
比例系数调整self.iou_ratio, self.iou_ratio代表iou的占比,(1-self.iou_ratio)为代表nwd的占比.
在TAL标签分配中使用:
在ultralytics/utils/tal.py中的def iou_calculation函数中进行更换即可.
以上这两可以配合使用,也可以单独使用.
定位损失系列(支持v8-detect、v8-seg、v8-pose、v10)
Powerful-IoU,Powerful-IoUV2,Inner-Powerful-IoU,Inner-Powerful-IoUV2,Focaler-Powerful-IoU,Focaler-Powerful-IoUV2论文链接 项目视频百度云链接-定位损失系列更换说明
Wise-IoU(v1,v2,v3)系列(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU,Powerful-IoU,Powerful-IoUV2).
Inner-Wise-IoU(v1,v2,v3)系列(IoU,WIoU,EIoU,GIoU,DIoU,CIoU,SIoU,MPDIoU,ShapeIoU,Powerful-IoU,Powerful-IoUV2). 项目视频百度云链接-20240111版本更新说明