YOLOv5 模型中的锚框使用指南

YOLOv5 模型使用多个检测头进行目标检测,每个检测头对应一个特征层。每个检测头都使用预定义的锚框来预测目标的尺寸和位置。

锚框定义:

anchors:
  - [6,6, 10,10, 14,15]  # P3/8
  - [18,19, 25,25, 32,32]  # P4/16
  - [40,41, 32,74, 59,62]  # P5/32
  - [110,47, 90,90, 133,137]  # P6/32

# 0-P1/2
# 1-P2/4
# 2-P3/8
# 3-P4/16
# 4-P5/32

示例:

假设你的网络输出到检测头的顺序是这样的:

[[22, 13, 16, 19], 1, 'Detect', ['nc', 'anchors']] 

其中,'22' 表示网络输出层序号。根据上面的锚框定义,第22层输出对应的是第2行锚框,也就是 P3/8 的锚框。

结论:

第22层输出到检测头使用的是 'P3/8' 的锚框。


原文地址: http://www.cveoy.top/t/topic/b3Xh 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录