Fix the input shape of anchors in ProposalLayer (matterport#961)

Fix the input shape of anchors in ProposalLayer and use "num_anchors" to replace "anchors" to avoid confusing
sirbastiano · Sep 28, 2018 · 1aca439 · 1aca439
1 parent d16c463
commit 1aca439
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mrcnn/model.py b/mrcnn/model.py
@@ -258,9 +258,9 @@ class ProposalLayer(KE.Layer):
     box refinement deltas to anchors.
 
     Inputs:
-        rpn_probs: [batch, anchors, (bg prob, fg prob)]
-        rpn_bbox: [batch, anchors, (dy, dx, log(dh), log(dw))]
-        anchors: [batch, (y1, x1, y2, x2)] anchors in normalized coordinates
+        rpn_probs: [batch, num_anchors, (bg prob, fg prob)]
+        rpn_bbox: [batch, num_anchors, (dy, dx, log(dh), log(dw))]
+        anchors: [batch, num_anchors, (y1, x1, y2, x2)] anchors in normalized coordinates
 
     Returns:
         Proposals in normalized coordinates [batch, rois, (y1, x1, y2, x2)]