From c4adda4835128db33b049b97b9f1a877b0d3a47a Mon Sep 17 00:00:00 2001
From: FDInSky <48318485+FDInSky@users.noreply.github.com>
Date: Tue, 16 Jul 2019 15:09:32 +0800
Subject: [PATCH] Fix  hard code in Cascade-RCNN (#2817)

---
 .../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 aad4b141..58936b3d 100644
--- a/PaddleCV/PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py
+++ b/PaddleCV/PaddleDetection/ppdet/modeling/roi_heads/cascade_head.py
@@ -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,
-- 
GitLab