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,