Make pre_nms_limit configurable

2ec88a35 · keineahnung2345 · Waleed · 9ca77d4a · 2ec88a35
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

mrcnn/config.py mrcnn/config.py +3 -0

未找到文件。
--- a/mrcnn/config.py
+++ b/mrcnn/config.py
@@ -90,6 +90,9 @@ class Config(object):

    # How many anchors per image to use for RPN training
    RPN_TRAIN_ANCHORS_PER_IMAGE = 256
+    
+    # ROIs kept after tf.nn.top_k and before non-maximum suppression
+    PRE_NMS_LIMIT = 6000

    # ROIs kept after non-maximum suppression (training and inference)
    POST_NMS_ROIS_TRAINING = 2000