fix ppyoloe negative batch assign (#7958)

ppdet/modeling/assigners/atss_assigner.py

@@ -126,7 +126,8 @@ class ATSSAssigner(nn.Layer):
             assigned_bboxes = paddle.zeros([batch_size, num_anchors, 4])
             assigned_scores = paddle.zeros(
                 [batch_size, num_anchors, self.num_classes])
-            return assigned_labels, assigned_bboxes, assigned_scores
+            mask_positive = paddle.zeros([batch_size, 1, num_anchors])
+            return assigned_labels, assigned_bboxes, assigned_scores, mask_positive
 
         # 1. compute iou between gt and anchor bbox, [B, n, L]
         ious = iou_similarity(gt_bboxes.reshape([-1, 4]), anchor_bboxes)

ppdet/modeling/assigners/task_aligned_assigner.py

@@ -120,7 +120,8 @@ class TaskAlignedAssigner(nn.Layer):
             assigned_bboxes = paddle.zeros([batch_size, num_anchors, 4])
             assigned_scores = paddle.zeros(
                 [batch_size, num_anchors, num_classes])
-            return assigned_labels, assigned_bboxes, assigned_scores
+            mask_positive = paddle.zeros([batch_size, 1, num_anchors])
+            return assigned_labels, assigned_bboxes, assigned_scores, mask_positive
 
         # compute iou between gt and pred bbox, [B, n, L]
         ious = batch_iou_similarity(gt_bboxes, pred_bboxes)

ppdet/modeling/assigners/task_aligned_assigner_cr.py

@@ -96,7 +96,8 @@ class TaskAlignedAssigner_CR(nn.Layer):
             assigned_bboxes = paddle.zeros([batch_size, num_anchors, 4])
             assigned_scores = paddle.zeros(
                 [batch_size, num_anchors, num_classes])
-            return assigned_labels, assigned_bboxes, assigned_scores
+            mask_positive = paddle.zeros([batch_size, 1, num_anchors])
+            return assigned_labels, assigned_bboxes, assigned_scores, mask_positive
 
         # compute iou between gt and pred bbox, [B, n, L]
         ious = batch_iou_similarity(gt_bboxes, pred_bboxes)