2.0 KiB
Loss系列
-
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)
-
IoU,GIoU,DIoU,CIoU,EIoU,SIoU,MPDIoU,ShapeIoU.
-
Inner-IoU,Inner-GIoU,Inner-DIoU,Inner-CIoU,Inner-EIoU,Inner-SIoU,Inner-ShapeIoU,Inner-MPDIoU.
-
Focaler-IoU系列(IoU,GIoU,DIoU,CIoU,EIoU,SIoU,WIoU,MPDIoU,ShapeIoU)
-
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版本更新说明
-