YOLO v8(学习笔记)

1.实验结果
  • 第一张图横轴参数量,第二张图横轴是一张图片推理的时间,这两个值都是越小越好
2.网络结构
(1)整体结构
  • 简化版
  • 细节版(这个没有区分Neck和Head)
  • 左边一列是Backbone的模块组成,每个模块可以在右边找到具体结构。
  • Neck部分其实就是PAN结构,和YOLO v5一样
  • Neck后面接的三个检测头(解耦头 ),每个检测头输出两个特征图,一个类别特征图,一个边框特征图。
(2)边框特征图输出参数(基于Anchor Free的输出)
  • 如上图,对于每个cell,64(16×4)维的输出,每16维对应一个方向的长度的概率,比如前16维是预测框左边距离中心点的长度,16维分别是长度取0、1、2.....16的概率,取最大概率的长度作为预测框左边距离中心点的长度,其余方向同理。
2.正负样本匹配
  • 1.获取三个检测头的输出结果(预测框、概率值)

  • 2.将三个检测头的结果映射到同一原图(640x640),同时将(l,t,r,b)坐标转化为左上坐标(Xmin,Ymin)和右下坐标(Xmax,Ymax),这样就可以确定一个预测框的位置了。

  • 3.初筛:所有的grid cell的中心点(anchor point)在GT框内的即为初始正样本(如上图)

  • 4.提取对应类别的pred score,计算CIOU计算align_matric=pred_scoreo.5 * Clou6根据align_matric的值,筛选出top-N作为正样本

  • 5.处理一个中心点可能匹配到多个GT框的情况,这时取CloU值大的作为对应GT的预测框。就是将中心点对应的预测框和GT1和GT2做CIOU,哪个大就作为那个GT的正样本。(上图)

4.损失函数
(1)损失组成
  • 预测框类别损失:BCE(二元交叉熵损失)

  • 预测框定位损失:CIoU、DFL(分布焦点损失)

  • LOSS = γ1Lcls + γ2Lciou + γ3Ldfl

(2)DFL
相关推荐
智者知已应修善业9 小时前
【4060BD_5V应用电路图】2022-12-24
驱动开发·经验分享·笔记·硬件架构·硬件工程
小弥儿9 小时前
GitHub今日热榜 | 2026-09-18:腾讯双项目上榜
学习·开源
知识分享小能手11 小时前
深度学习学习教程,从入门到精通,深度生成模型 —— 知识点详解与代码实现(20)
人工智能·深度学习·学习
m4Rk_12 小时前
【论文阅读】Agent 记忆机制(74):CompassMem——从相似度检索走向事件图上的记忆导航
论文阅读·人工智能·学习·开源·github
小李不想当小白12 小时前
DMA直接存储器存取(STM32标准库学习笔记)
笔记·stm32·单片机·嵌入式硬件·学习·分享
2601_9499506314 小时前
练题簿在线练题全流程:从资料导入到模拟考试与错题复盘
学习·考研·小程序·刷题·小程序推荐
2601_9653842614 小时前
口腔黏膜清洁的工程化路径:从表面形貌表征到清洁策略匹配
经验分享·笔记
泡泡鱼(敲代码中)14 小时前
Python 容器类型学习笔记:列表、元组、字典、集合
开发语言·笔记·python·pycharm
别动我齐刘海14 小时前
ROS2 Jazzy + C++ 实战路线——基础学习2
c++·人工智能·vscode·python·学习·机器学习·机器人
Hotchip_MEMS16 小时前
传统咪头与MEMS硅麦:雾化器气流传感方案对比
人工智能·笔记·物联网·电脑·制造