yolov6-10算法梳理(主要为head部分的设计)

yolov6

年份:2022
机构:美团
文章:https://arxiv.org/pdf/2301.05586
代码:https://github.com/meituan/YOLOv6
在这里插入图片描述在这里插入图片描述在这里插入图片描述

  • Label assignment
    Warmup用ATSS,之后用TAL
  • Class loss
    VariFocal Loss
  • object loss
  • 加了object loss会变差(会改变TAL计算逻辑)
  • Reg loss
    SIoU GIoU
  • 亮点
    1,自蒸馏,进一步提升效果,大模型和小模型用不同策略
    2,因为马赛克增强的原因,测试时添加灰边能更好检测边界目标。训练后期关闭马赛克增强并用宽度更小的灰边会好点
    3,训练时增加anchor base的辅助头,推理时去掉,anchor free较于base速度更快,这样训练可能爆显存
    4,量化
    5,重参数

yolov7

年份:2022
作者:Chien-Yao Wang,Alexey Bochkovskiy
文章:https://arxiv.org/pdf/2207.02696
代码:https://github.com/WongKinYiu/yolov7
在这里插入图片描述

  • Label assignment
    SimOTA
  • Class loss
    BCE loss
  • object loss
    BCE loss iou值作为软标签(好像是obj学前后景2分类,cls不学背景,anchor base版的yolox)
  • Reg loss
    IoU loss
  • 亮点
  • 1,anchor base可能对小目标更友好
  • 2,添加辅助头分配更多正样本,标签匹配由lead head主导
  • 3,参考yolor添加隐式和显式知识学习
  • 4,重参数

yolov8

年份:2023
机构:Ultralytics
代码:https://github.com/ultralytics/ultralytics
在这里插入图片描述

  • Label assignment
    TAL
  • Class loss
    VariFocal Loss
  • object loss
  • Reg loss
    DFL loss 和CIOU loss
  • 亮点
  • 1,DFL loss,TAL+VariFocal Loss好用

yolov9

年份:2024
作者:Chien-Yao Wang
文章:https://arxiv.org/pdf/2402.13616
代码:https://github.com/WongKinYiu/yolov9
在这里插入图片描述

  • Label assignment
    TAL
  • Class loss
    BCE
  • object loss
  • Reg loss
    DFL loss 和IOU loss
  • 亮点
  • 1,添加辅助分支降低信息损失

yolov10

年份:2024
机构:清华
文章:https://arxiv.org/pdf/2405.14458
代码:https://github.com/THU-MIG/yolov10
在这里插入图片描述

  • Label assignment
    TAL
  • Class loss
    BCE
  • object loss
  • Reg loss
    DFL loss 和IOU loss
  • 亮点
  • 1,one2one(tal topk=1)和one2many(tal topk=10)两个head训练,推理只用one2one
  • 2,轻量的cls head,复杂度更低的下采样,更紧凑的block设计
  • 3,深层用大核卷积和注意力机制
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值