Fix NMS settings for evaluation / postprocessing (#278)

f982b5bf · Martin Hahner · GitHub · 9e460da2 · f982b5bf
Unverified Commit f982b5bf authored Sep 08, 2020 by Martin Hahner Committed by GitHub Sep 08, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/cfgs/kitti_models/second_multihead.yaml tools/cfgs/kitti_models/second_multihead.yaml +2 -2

No files found.
--- a/tools/cfgs/kitti_models/second_multihead.yaml
+++ b/tools/cfgs/kitti_models/second_multihead.yaml
@@ -101,14 +101,14 @@ MODEL:

    POST_PROCESSING:
        RECALL_THRESH_LIST: [0.3, 0.5, 0.7]
-        MULTI_CLASSES_NMS: False
+        MULTI_CLASSES_NMS: True
        SCORE_THRESH: 0.1
        OUTPUT_RAW_SCORE: False

        EVAL_METRIC: kitti

        NMS_CONFIG:
-            MULTI_CLASSES_NMS: False
+            MULTI_CLASSES_NMS: True
            NMS_TYPE: nms_gpu
            NMS_THRESH: 0.1
            NMS_PRE_MAXSIZE: 4096