update ops (#5443)

ppdet/modeling/heads/mask_head.py

@@ -209,7 +209,7 @@ class MaskHead(nn.Layer):
         scale_factor (Tensor): The scale factor from origin size to input size
         """
         if not self.export_onnx and rois.shape[0] == 0:
-            mask_out = paddle.full([1, 1, 1, 1], -1)
+            mask_out = paddle.full([1, 1, 1], -1)
         else:
             bbox = [rois[:, 2:]]
             labels = rois[:, 0].cast('int32')

ppdet/modeling/ops.py

@@ -482,6 +482,8 @@ def collect_fpn_proposals(multi_rois,
             rois_num = helper.create_variable_for_type_inference(dtype='int32')
             rois_num.stop_gradient = True
             outputs['RoisNum'] = rois_num
+        else:
+            rois_num = None
         helper.append_op(
             type='collect_fpn_proposals',
             inputs=inputs,
@@ -602,6 +604,8 @@ def distribute_fpn_proposals(fpn_rois,
                 for i in range(num_lvl)
             ]
             outputs['MultiLevelRoIsNum'] = rois_num_per_level
+        else:
+            rois_num_per_level = None
 
         helper.append_op(
             type='distribute_fpn_proposals',