未验证 提交 e50ca5b3 编写于 作者: L LokeZhou 提交者: GitHub

tood model fix assigner_bug (#7897)

上级 03421ea4
...@@ -293,7 +293,7 @@ class TOODHead(nn.Layer): ...@@ -293,7 +293,7 @@ class TOODHead(nn.Layer):
pad_gt_mask = gt_meta['pad_gt_mask'] pad_gt_mask = gt_meta['pad_gt_mask']
# label assignment # label assignment
if gt_meta['epoch_id'] < self.static_assigner_epoch: if gt_meta['epoch_id'] < self.static_assigner_epoch:
assigned_labels, assigned_bboxes, assigned_scores, _ = self.static_assigner( assigned_labels, assigned_bboxes, assigned_scores = self.static_assigner(
anchors, anchors,
num_anchors_list, num_anchors_list,
gt_labels, gt_labels,
...@@ -302,7 +302,7 @@ class TOODHead(nn.Layer): ...@@ -302,7 +302,7 @@ class TOODHead(nn.Layer):
bg_index=self.num_classes) bg_index=self.num_classes)
alpha_l = 0.25 alpha_l = 0.25
else: else:
assigned_labels, assigned_bboxes, assigned_scores, _ = self.assigner( assigned_labels, assigned_bboxes, assigned_scores = self.assigner(
pred_scores.detach(), pred_scores.detach(),
pred_bboxes.detach() * stride_tensor, pred_bboxes.detach() * stride_tensor,
bbox_center(anchors), bbox_center(anchors),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册