yolo11-seg模型后处理输出节点

检测头节点分析总结

1. Bounding Box 头 (cv2) - 回归头部

  • /model.23/cv2.0/cv2.0.2/Conv_output_0: 1, 64, 80, 80

  • /model.23/cv2.1/cv2.1.2/Conv_output_0: 1, 64, 40, 40

  • /model.23/cv2.2/cv2.2.2/Conv_output_0: 1, 64, 20, 20

作用: 输出边界框回归参数,64个通道对应4个坐标×16个DFL(Distribution Focal Loss)bins,用于精确的边界框定位。

2. 分类头 (cv3) - 类别预测头部 7类

  • /model.23/cv3.0/cv3.0.2/Conv_output_0: 1, 7, 80, 80

  • /model.23/cv3.1/cv3.1.2/Conv_output_0: 1, 7, 40, 40

  • /model.23/cv3.2/cv3.2.2/Conv_output_0: 1, 7, 20, 20

作用: 输出7个类别的概率分数,每个空间位置对应7个类别的置信度。

3. 分割头 (cv4) - 掩码系数头部

  • /model.23/cv4.0/cv4.0.2/Conv_output_0: 1, 32, 80, 80

  • /model.23/cv4.1/cv4.1.2/Conv_output_0: 1, 32, 40, 40

  • /model.23/cv4.2/cv4.2.2/Conv_output_0: 1, 32, 20, 20

作用: 输出32个掩码系数,用于生成实例分割掩码,这三个不同尺度的输出会被合并。

数据流向:

  1. cv2和cv3头 → 通过Concat操作合并到最终的output0输出
  • cv2输出经过DFL处理链(Reshape→Transpose→Softmax→Conv→Slice→数学运算)→最终bbox坐标

  • cv3输出经过Sigmoid激活→最终类别概率

  1. cv4头 → 经过Reshape和Concat操作形成掩码系数 → 存储在output0的最后32个通道中

  2. 独立的分割路径 → 通过proto网络分支生成output1

  • 来源:来自主干网络的特征

  • 处理:Conv→ReLU→ConvTranspose(上采样)→Conv→ReLU→Conv→ReLU

  • 输出:1, 32, 160, 160 → 直接作为output1输出

多尺度检测策略:

  • Scale 0 (大物体): 80×80网格,适合检测较大目标

  • Scale 1 (中物体): 40×40网格,适合检测中等目标

  • Scale 2 (小物体): 20×20网格,适合检测较小目标

特征层 典型输入尺寸(以640×640图为例) 主要任务
P3 1, 128, 80, 80 检测小目标(<32×32像素)
P4 1, 256, 40, 40 检测中等目标(32--96像素)
P5 1, 512, 20, 20 检测大目标(>96像素)

对应的rknn模型图:

相关推荐
哥不是小萝莉几秒前
AI 编码 Agent 从原理到可运行代码
ai·agent
图王大胜13 分钟前
万物演化论00(序章) 从宇宙到AI
人工智能·ai·宇宙·演化·文明·生命科学
技术小事1 小时前
How To Use Claude 怎么使用Claude
ai·claude
笑霸final1 小时前
不用上传服务器!用 Vue3 + ONNX Runtime Web 在浏览器本地实现智能抠图
运维·前端·图像处理·ai·onnx·canva可画·web性能优化
一切皆是因缘际会1 小时前
多时空虚拟中心—— 全域同源基座与脑机谐振应用架构
ai·系统架构·脑机接口·虚拟全球网络控制中心·元宇宙世界底层架构·原生多时空架构
效率工作实验室2 小时前
AI Agent开发平台有哪些?2026年平台对比
ai·agent开发平台·ai智能体开发平台·智能体平台对比·agent平台选型
尘中远3 小时前
7大开源Agent源码对比解读——上下文管理
ai·开源·agent·codex·harness
Elastic 中国社区官方博客3 小时前
从建议到修复的 4 个阶段:使用 Elastic Workflows 实现人在回路中的自动化
运维·数据库·人工智能·后端·elasticsearch·ai·自动化
土星云SaturnCloud3 小时前
大型仓储AI视觉全场景落地实战:安全·效率·库存,32TOPS土星云边缘算力赋能分区部署
服务器·人工智能·ai·边缘计算
电子工匠3 小时前
Altium + Quilter 搭建指南
嵌入式硬件·ai·pcb工艺