目标检测网络锚框配置详解

目标检测网络锚框配置详解

本文将解释如何根据目标检测网络的输出层和预定义的锚框列表确定使用哪组锚框进行目标检测。

锚框列表：

• [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层输出到检测头使用的是第2行锚框（索引为1）。
• 13层输出到检测头使用的是第0行锚框（索引为0）。

解释:

在网络输出到检测头的顺序中，[22, 13, 16, 19] 表示对应输出层的索引。根据网络输出层索引和锚框列表的对应关系，我们可以找到对应的锚框。例如，22层的索引为2，对应的是 P3/8 的锚框，即 [6,6, 10,10, 14,15]。

因此，在目标检测中，我们需要根据网络输出层索引和预定义的锚框列表来确定使用哪组锚框进行目标检测。