YOLOV8/11分割与分类输出参数说明

YOLOV8/11 输出三个尺度的特征图

对于640x640图像输入:

  1. 80x80 检测小目标
  2. 40x40 检测中目标
  3. 20x20 检测大目标

检测输出说明

对于640x640的图片输入,输出output084,8400,预测框的总数量是8400,每个预测框的维度是84(4+80),针对COCO数据集的80个类别而言

8400x84 = 80x80x84+40x40x84+20x20x84 = 80x80x(4+80)+40x40x(4+80)+20x20x(4+80)

其中4为预测框 cx,cy,w,h,代表预测框的中心点坐标与宽高;80为80个类别的置信度

输入图像大小不一样,输出的检测框数量不同

分割输出说明

分割有两个输出output0与output1,对于640x640 的图像输入,output0为检测头的输出其维度1x116x8400,output1为分割头的输出,其维度为1x32x160x160

output0维度1x116x8400,代表输出的检测框数量为8400,每个预测框的维度为116(4+80+32)

其中4为预测框 cx,cy,w,h,代表预测框的中心点坐标与宽高;80为80个类别的置信度,32维向量为与当前预测框关联的分割mask系数

output1分割头的输出维度为1x32x160x160,代表32个尺寸为160x160的基础mask,这些基础mask所有检测框共享,每个检测框的32维mask系数与output1做矩阵乘法得到当前检测框的最终mask

具体实现:

  1. Mask 系数(n×32) × 模型输出原型output1(32×160×160),n为筛选后预测框的数量
  2. 经过 sigmoid 激活
  3. 得到 n×160×160 的原始 Mask
  4. 把框从 640×640 → 缩放到 160×160,用于裁剪 Mask
  5. 把检测框以外的 Mask 全部设为 0
  6. 只保留框内的物体区域
  7. 把 160×160 中多余的灰色区域切掉(对于输入letterbox),得到有效小图
  8. 双线性插值缩放,把 Mask 缩放到 原图分辨率(如 1920×1080)
相关推荐
前沿在线4 分钟前
2026 WRC世界机器人大会| 黑芝麻智能展台亮点
人工智能·ai·大模型
QYRdata10 分钟前
28.7%年复合增速锚定AI模型安全赛道,2026-2032年行业步入高速扩容新阶段
人工智能·安全
皮卡丘不断更12 分钟前
Vibe Coding 做完原型后:给网页项目加上任务、验证和人工确认
人工智能·软件工程
Leo.yuan18 分钟前
FineDataLink 5.0 vs Great Expectations:数据质量治理工具深度横评,国产低代码方案能否替代开源标杆?
大数据·人工智能
薛定e的猫咪19 分钟前
(arXiv 2026)GLiBRL :可学习基函数的深度贝叶斯元强化学习 ----待补充
人工智能·深度学习·学习·算法·机器学习
喜欢吃豆20 分钟前
GEO 到底怎么做?从 Prompt 研究到 AI Citation 的完整落地方法
人工智能·prompt
yingyuecom21 分钟前
映悦AI × Joverse全球发布:四大重磅更新,AI创作进入工业化时代
人工智能·gpt·chatgpt·prompt·aigc
会周易的程序员27 分钟前
软件接入大模型实现 Agent —— 从原理到 C++ 落地完全指南
c++·人工智能·物联网·架构·agent·工业协议·mcp
躺柒29 分钟前
读数据可视化13空间标量场(上)
人工智能·深度学习·信息可视化·数据可视化·空间·大数据分析
小王20432 分钟前
Day 28:目标检测入门 — 两阶段 vs 单阶段
人工智能·目标检测·计算机视觉