#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
没有参考框时,不读取参考框数据,两个候选框都判为负样本。