mmdet——SamplingResult索引对应解析

例子

重要属性

所有和​​pred​​相关排列顺序都是以预测值的​​idx​​顺序重新排列，GT都是按照默认顺序排列，按照GT的索引拿到对应的GT。

• ​​pos_assigned_gt_inds​​​和​​pos_inds​​是一一对应的：

• 第3个​​gt​​​对应第12个​​pred​​​，第0个​​gt​​​对应第25个​​pred​​​，第2个​​gt​​​对应第38个​​pred​​​，第1个​​gt​​​对应第48个​​pred​​。

pos_assigned_gt_inds = tensor([3, 0, 2, 1], device='cuda:0')
pos_inds = tensor([12, 25, 38, 48], device='cuda:0')  # 经过sort，默认按照pred顺序排列

• ​​pos_gt_bboxes​​​和​​pos_bboxes​​​是经过排序后已经一一对应的两个​​tensor​​​，此时不需要管其他索引，第一个​​pos_gt_bboxes​​​就对应第一个​​pos_bboxes​​元素。

这里的​​gt_boxes​​​为函数的输入，而​​SamplingResult​​​中的​​pos_gt_bboxes​​​是已经按照​​pos_assigned_gt_inds​​的顺序排列过的GT。