We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a2f2476 commit 576f52cCopy full SHA for 576f52c
ppdet/modeling/proposal_generator/target.py
@@ -50,8 +50,8 @@ def rpn_anchor_target(anchors,
50
labels = paddle.scatter(labels, fg_inds, paddle.ones_like(fg_inds))
51
# Step3: make output
52
if gt_bbox.shape[0] == 0:
53
- matched_gt_boxes = paddle.zeros([0, 4])
54
- tgt_delta = paddle.zeros([0, 4])
+ matched_gt_boxes = paddle.zeros([matches.shape[0], 4])
+ tgt_delta = paddle.zeros([matches.shape[0], 4])
55
else:
56
matched_gt_boxes = paddle.gather(gt_bbox, matches)
57
tgt_delta = bbox2delta(anchors, matched_gt_boxes, weights)
0 commit comments