update pointrcnn_roiawarepool.yaml

hcheng1005 · Jul 23, 2020 · 48400f1 · 48400f1
1 parent 29ad7c8
commit 48400f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/cfgs/kitti_models/pointrcnn_roiawarepool.yaml b/tools/cfgs/kitti_models/pointrcnn_roiawarepool.yaml
@@ -81,9 +81,9 @@ MODEL:
             TEST:
                 NMS_TYPE: nms_gpu
                 MULTI_CLASSES_NMS: False
-                NMS_PRE_MAXSIZE: 1024
+                NMS_PRE_MAXSIZE: 9000
                 NMS_POST_MAXSIZE: 100
-                NMS_THRESH: 0.7
+                NMS_THRESH: 0.85
 
         ROI_AWARE_POOL:
             POOL_SIZE: 12