From 1aca439c37849dcd085167c4e69d3abcd9d368d7 Mon Sep 17 00:00:00 2001 From: keineahnung2345 Date: Sat, 29 Sep 2018 07:54:59 +0800 Subject: [PATCH] Fix the input shape of anchors in ProposalLayer (#961) Fix the input shape of anchors in ProposalLayer and use "num_anchors" to replace "anchors" to avoid confusing --- mrcnn/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mrcnn/model.py b/mrcnn/model.py index 7ecb860a07..72ee65f566 100644 --- 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)]