From c399ca4420ec552f79ebd5867e2f7ca483cfd5ab Mon Sep 17 00:00:00 2001 From: FDInSky <48318485+FDInSky@users.noreply.github.com> Date: Tue, 16 Jul 2019 16:15:00 +0800 Subject: [PATCH] Fix hard code in Cascade-RCNN head (#2818) --- .../PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleCV/PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py b/PaddleCV/PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py index 8971b4e0..56327940 100644 --- a/PaddleCV/PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py +++ b/PaddleCV/PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py @@ -196,7 +196,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, -- GitLab