Skip to content

Commit

Permalink
update (#6926) (#6927)
Browse files Browse the repository at this point in the history
  • Loading branch information
lyuwenyu authored Sep 13, 2022
1 parent 8a1efe2 commit 443e4a3
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 2 deletions.
1 change: 0 additions & 1 deletion configs/vitdet/cascade_rcnn_vit_base_hrfpn_cae_1x_coco.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
2 changes: 1 addition & 1 deletion ppdet/modeling/heads/roi_extractor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 443e4a3

Please sign in to comment.