建站网站加盟,案例模板我的网站,wordpress thefox,wordpress国外主题慢参考文章#xff1a; 1、YOLOv5 深度剖析 2、如何看待YOLOv8#xff0c;YOLOv5作者开源新作#xff0c;它来了#xff01;? 3、anchor的简单理解
完整网络结构 YOLO v5和YOLO v8的Head部分 YOLO v8的Head 部分相比 YOLOv5 改动较大#xff0c;换成了目前主流的解耦头结构…参考文章 1、YOLOv5 深度剖析 2、如何看待YOLOv8YOLOv5作者开源新作它来了? 3、anchor的简单理解
完整网络结构 YOLO v5和YOLO v8的Head部分 YOLO v8的Head 部分相比 YOLOv5 改动较大换成了目前主流的解耦头结构将分类和检测头分离同时也从 Anchor-Based 换成了 Anchor-Free。anchor-based和anchor-free方法的本质区别就是如何定义正负样本。 Anchor-Based事先通过手工或聚类方法设定好的具有不同尺寸、宽高比的方框anchor_boxes。这些方框覆盖了整张图像目的是为了防止漏检。在模型训练过程中根据anchor_boxes与ground truth的IoU交并比损失对anchor_boxes的长宽以及位置进行回归使其越来越接近ground truth在回归的同时预测anchor_boxes的类别最终输出这些回归分类好的anchors。目前主流的目标检测算法多是anchor-based这一类其中有two-stage也有one-stage。two-stage方法要筛选和优化的anchors数量要远超one-stage方法筛选步骤较为严谨所以耗费时间要久一些但是精度要高一些。在常用的检测基准上SOTA的方法一般都是anchor-based的。 Anchor-Free由于FPN和Focal Loss的出现有效解决了特征语义信息不足和正负样本不均衡的问题又涌现出了一批anchor-free的算法。anchor-free检测器以两种不同的方式检测物体一种是首先定位到多个预定义或自学习的关键点然后约束物体的空间范围称为Keypoint-based方法另一种是利用中心点或中心目标区域来定义正样本然后预测其到目标四个边的距离称为Center-based方法。anchor-free的方法使得目标检测的流程进一步精简减少了相关超参数使得网络搭建训练更简便泛化能力也更强。
YOLOv5的C3结构 和 YOLOv8的C2f结构 YOLO v8将 YOLOv5 的 C3 结构换成了梯度流更丰富的 C2f 结构并对不同尺度模型调整了不同的通道数不过这个 C2f 模块中存在 Split 等操作对特定硬件部署没有之前那么友好了。
YOLO v5和YOLO v8的loss
1、YOLO v5在类别的损失函数中使用的是二元交叉熵损失函数而不是softmax损失函数。因为softmax损失函数对于每个类别的处理是互斥的。而yolo考虑到了类别不互斥的情况例如一个类别可以是车但也可以是汽车或货车如果用softmax是车就不可以是汽车但二元交叉熵既可以是车也可以是汽车。 2、YOLO v8采用了 TaskAlignedAssigner 正样本分配策略并引入了 Distribution Focal Loss。Loss 计算包括 2 个分支 分类和回归分支没有了之前的 objectness 分支。分类分支依然采用 BCE Loss。回归分支需要和 Distribution Focal Loss 中提出的积分形式表示法绑定因此使用了 Distribution Focal Loss 同时还使用了 CIoU Loss。3 个 Loss 采用一定权重比例加权即可。 TaskAlignedAssigner 正样本分配策略核心思想是在候选框与真实目标之间建立正确的对应关系以便在训练过程中能够准确地学习目标检测模型即根据分类与回归的分数加权的分数选择正样本。在目标检测中类别不平衡是指不同类别的目标数量差异很大这可能会导致模型在训练过程中对少数类别的预测效果较差。Focal Loss 通过在损失函数中引入一个可调节的因子来减小易分类样本的权重从而集中训练模型在困难样本上。Distribution Focal Loss是 Focal Loss 的一个变体专门用于应对类别不平衡问题。其主要思想是将 Focal Loss 与样本分布信息相结合从而更好地处理不平衡的类别。具体而言分布 Focal 损失会根据每个类别的样本分布情况来调整 Focal Loss 中的因子以便更有效地关注罕见类别。
YOLO v5和YOLO v8训练数据增强
YOLO v8数据增强方面和 YOLOv5 差距不大只不过引入了 YOLOX 中提出的最后 10 个 epoch 关闭 Mosaic 的操作。假设训练 epoch 是 500其示意图如下所示 Mosaic是YOLOV4中提出的新方法适用于目标检测主要思想是将四张图片进行拼接到一张图上作为训练样本。由于Mosaic用于目标检测进行拼接时目标框的坐标也要做相应的变化。Mosaic的主要操作如下
对每一张图进行随机裁剪得到Acrop的目标是选择原图的某一块区域而不是全图进行后续的拼接将A进行resize到输出图大小得到B目标是统一坐标系尺寸方便后续拼接坐标框会进行相应的缩放将B随机裁剪一块指定大小的区域C将C粘贴到输出图相应的位置此时只需要坐标平移就可以调整坐标框
