提交 af8b1b99 编写于 作者: J jerrywgz 提交者: qingqing01

Fix rpn_target_assign API.

上级 f6f5cdaa
...@@ -164,7 +164,7 @@ def rpn_target_assign(loc, ...@@ -164,7 +164,7 @@ def rpn_target_assign(loc,
}) })
# 4. Reshape and gather the target entry # 4. Reshape and gather the target entry
scores = nn.reshape(x=scores, shape=(-1, 1)) scores = nn.reshape(x=scores, shape=(-1, 2))
loc = nn.reshape(x=loc, shape=(-1, 4)) loc = nn.reshape(x=loc, shape=(-1, 4))
target_label = nn.reshape(x=target_label, shape=(-1, 1)) target_label = nn.reshape(x=target_label, shape=(-1, 1))
target_bbox = nn.reshape(x=target_bbox, shape=(-1, 4)) target_bbox = nn.reshape(x=target_bbox, shape=(-1, 4))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册