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

Fix hard code in Cascade-RCNN (#2817)

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