update (#6926) (#6927)

PaddlePaddle · Sep 13, 2022 · 443e4a3 · 443e4a3
1 parent 8a1efe2
commit 443e4a3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml b/configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml
@@ -98,7 +98,6 @@ CascadeHead:
   reg_class_agnostic: False
   stage_loss_weights: [1, 0.5, 0.25]
   loss_normalize_pos: True
-  add_gt_as_proposals: [True, True, True]
 
 
 BBoxAssigner:

diff --git a/ppdet/modeling/heads/roi_extractor.py b/ppdet/modeling/heads/roi_extractor.py
@@ -96,7 +96,7 @@ def __call__(self, feats, roi, rois_num):
                     self.canonical_size,
                     rois_num=rois_num)
             else:
-                ops.distribute_fpn_proposals(
+                rois_dist, restore_index, rois_num_dist = ops.distribute_fpn_proposals(
                     roi,
                     k_min,
                     k_max,