Skip to content

Commit af8b1b9

Browse files
jerrywgzqingqing01
authored andcommitted
Fix rpn_target_assign API.
1 parent f6f5cda commit af8b1b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/fluid/layers/detection.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ class number, M is number of bounding boxes. For each category
164164
})
165165

166166
# 4. Reshape and gather the target entry
167-
scores = nn.reshape(x=scores, shape=(-1, 1))
167+
scores = nn.reshape(x=scores, shape=(-1, 2))
168168
loc = nn.reshape(x=loc, shape=(-1, 4))
169169
target_label = nn.reshape(x=target_label, shape=(-1, 1))
170170
target_bbox = nn.reshape(x=target_bbox, shape=(-1, 4))

0 commit comments

Comments
 (0)