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。