Fix rpn_target_assign API.

7 years ago · af8b1b993f
parent f6f5cdaa05
commit af8b1b993f
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/detection.py
+++ b/python/paddle/fluid/layers/detection.py
@ -164,7 +164,7 @@ def rpn_target_assign(loc,
        })

    # 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))
    target_label = nn.reshape(x=target_label, shape=(-1, 1))
    target_bbox = nn.reshape(x=target_bbox, shape=(-1, 4))