From 12e76a1794ea73922b8c21d5b885e49dde599327 Mon Sep 17 00:00:00 2001 From: Vladyslav Date: Thu, 10 Jan 2019 14:35:44 +0800 Subject: [PATCH] Fix typo in rpn_class_loss_graph Fix https://github.com/matterport/Mask_RCNN/issues/870 --- mrcnn/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mrcnn/model.py b/mrcnn/model.py index 9d23f38..99ea633 100644 --- a/mrcnn/model.py +++ b/mrcnn/model.py @@ -1025,7 +1025,7 @@ def rpn_class_loss_graph(rpn_match, rpn_class_logits): rpn_match: [batch, anchors, 1]. Anchor match type. 1=positive, -1=negative, 0=neutral anchor. - rpn_class_logits: [batch, anchors, 2]. RPN classifier logits for FG/BG. + rpn_class_logits: [batch, anchors, 2]. RPN classifier logits for BG/FG. """ # Squeeze last dim to simplify rpn_match = tf.squeeze(rpn_match, -1)