From 443e4a313ed28a07a33ef65b6c35fa732b1148bb Mon Sep 17 00:00:00 2001 From: Wenyu Date: Tue, 13 Sep 2022 14:26:36 +0800 Subject: [PATCH] update (#6926) (#6927) --- configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml | 1 - ppdet/modeling/heads/roi_extractor.py | 2 +- 2 files changed, 1 insertion(+), 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 index 081e371e209..c27b4e6c417 100644 --- 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 index 19ce3df826f..a79a7fd1512 100644 --- 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,