未验证 提交 dd2d5795 编写于 作者: F Feng Ni 提交者: GitHub

[cherry-pick] fix tal assigner cr (#7731)

上级 f5394864
......@@ -178,4 +178,4 @@ class TaskAlignedAssigner_CR(nn.Layer):
alignment_metrics = alignment_metrics.max(-2).unsqueeze(-1)
assigned_scores = assigned_scores * alignment_metrics
return assigned_labels, assigned_bboxes, assigned_scores
return assigned_labels, assigned_bboxes, assigned_scores, mask_positive
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册