From dd2d57955a1552c68cec8eb3a77f9bfa8537afc6 Mon Sep 17 00:00:00 2001 From: Feng Ni Date: Fri, 10 Feb 2023 18:48:24 +0800 Subject: [PATCH] [cherry-pick] fix tal assigner cr (#7731) --- ppdet/modeling/assigners/task_aligned_assigner_cr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/assigners/task_aligned_assigner_cr.py b/ppdet/modeling/assigners/task_aligned_assigner_cr.py index 5c5097604..4558d6e8e 100644 --- a/ppdet/modeling/assigners/task_aligned_assigner_cr.py +++ b/ppdet/modeling/assigners/task_aligned_assigner_cr.py @@ -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 -- GitLab