diff --git a/tools/cfgs/kitti_models/pointrcnn_roiawarepool.yaml b/tools/cfgs/kitti_models/pointrcnn_roiawarepool.yaml index 25ad7039d..6b8d45ca0 100644 --- 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