#AEAI03004. 候选框监督标签分配

候选框监督标签分配

题目描述

训练检测器时,候选框需要根据与标注框的重叠程度分成正样本、负样本或忽略样本。具身智能部将这个分配过程抽象成一次 IoU 计算任务。

给定 M 个参考框和 N 个候选框。对每个候选框,找到 IoU 最大的参考框,根据两个阈值输出标签:正样本 P reference_id、负样本 N,或忽略样本 I

输入格式

  • 第一行输入两个整数 M N
  • 随后 M 行输入参考框 x1 y1 x2 y2
  • 再随后 N 行输入候选框 x1 y1 x2 y2
  • 最后一行输入两个浮点数 Tpos Tneg

输出格式

  • 对每个候选框输出一行。
  • 若为正样本,输出 P reference_id
  • 若为负样本,输出 N
  • 否则输出 I

判定规则

使用标准输出比较。

  • 若没有参考框,所有候选框均输出 N
  • maxIoU >= Tpos 为正样本。
  • maxIoU < Tneg 为负样本。
  • 其余情况为忽略样本。
  • 多个参考框 IoU 并列最大时,取编号较小的参考框。参考框编号从 1 开始。

样例

输入样例 1

2 3
0 0 10 10
20 20 30 30
1 1 9 9
5 5 15 15
40 40 50 50
0.5 0.2

输出样例 1

P 1
N
N

样例解释 1

第一个候选框与第一个参考框 IoU 达到正样本阈值;后两个候选框的最大 IoU 均小于负样本阈值。

输入样例 2

0 2
0 0 1 1
2 2 3 3
0.5 0.1

输出样例 2

N
N

样例解释 2

没有参考框时,不读取参考框数据,两个候选框都判为负样本。