提交 3371a8c5 编写于 作者: F FDInSky 提交者: qingqing01

Fix hard code in Cascade-RCNN (#2817)

上级 2953d108
......@@ -197,7 +197,7 @@ class CascadeBBoxHead(object):
# only use fg box delta to decode box
bbox_pred_new = fluid.layers.slice(
bbox_pred_new, axes=[1], starts=[1], ends=[2])
bbox_pred_new = fluid.layers.expand(bbox_pred_new, [1, 81, 1])
bbox_pred_new = fluid.layers.expand(bbox_pred_new, [1, self.num_classes, 1])
decoded_box = fluid.layers.box_coder(
prior_box=proposals_boxes,
prior_box_var=bbox_reg_w,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册