提交 b29b32c7 编写于 作者: keineahnung2345's avatar keineahnung2345 提交者: Waleed

Fix typo in rpn_class_loss_graph

Fix https://github.com/matterport/Mask_RCNN/issues/870
上级 1721e841
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册