From 367946e7c8cb93460e9e3cd4b2ce14f37c6c8815 Mon Sep 17 00:00:00 2001 From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com> Date: Mon, 15 Nov 2021 11:31:51 +0800 Subject: [PATCH] fix assign error (#4533) --- ppdet/modeling/proposal_generator/target_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/proposal_generator/target_layer.py b/ppdet/modeling/proposal_generator/target_layer.py index e6abb60f0..3b5a09601 100644 --- a/ppdet/modeling/proposal_generator/target_layer.py +++ b/ppdet/modeling/proposal_generator/target_layer.py @@ -469,7 +469,7 @@ class RBoxAssigner(object): bbox_targets = np.zeros_like(anchors) bbox_weights = np.zeros_like(anchors) bbox_gt_bboxes = np.zeros_like(anchors) - pos_labels = np.ones(anchors_num, dtype=np.int32) * -1 + pos_labels = np.zeros(anchors_num, dtype=np.int32) pos_labels_weights = np.zeros(anchors_num, dtype=np.float32) pos_sampled_anchors = anchors[pos_inds] -- GitLab