support DISABLE_AUG_LIST config for using --set disable augs

pcdet/datasets/augmentor/data_augmentor.py

@@ -11,7 +11,13 @@ class DataAugmentor(object):
         self.logger = logger
 
         self.data_augmentor_queue = []
-        for cur_cfg in augmentor_configs:
+        aug_config_list = augmentor_configs if isinstance(augmentor_configs, list) \
+            else augmentor_configs.AUG_CONFIG_LIST
+
+        for cur_cfg in aug_config_list:
+            if not isinstance(augmentor_configs, list):
+                if cur_cfg.NAME in augmentor_configs.DISABLE_AUG_LIST:
+                    continue
             cur_augmentor = getattr(self, cur_cfg.NAME)(config=cur_cfg)
             self.data_augmentor_queue.append(cur_augmentor)
 

tools/cfgs/dataset_configs/kitti_dataset.yaml

@@ -17,6 +17,8 @@ FOV_POINTS_ONLY: True
 
 
 DATA_AUGMENTOR:
+    DISABLE_AUG_LIST: ['placeholder']
+    AUG_CONFIG_LIST:
         - NAME: gt_sampling
           USE_ROAD_PLANE: True
           DB_INFO_PATH:

tools/cfgs/kitti_models/pointpillar.yaml

@@ -21,6 +21,8 @@ DATA_CONFIG:
             'test': 40000
           }
     DATA_AUGMENTOR:
+        DISABLE_AUG_LIST: ['placeholder']
+        AUG_CONFIG_LIST:
             - NAME: gt_sampling
               USE_ROAD_PLANE: True
               DB_INFO_PATH:

tools/cfgs/kitti_models/pv_rcnn.yaml

@@ -3,6 +3,8 @@ CLASS_NAMES: ['Car', 'Pedestrian', 'Cyclist']
 DATA_CONFIG:
     _BASE_CONFIG_: cfgs/dataset_configs/kitti_dataset.yaml
     DATA_AUGMENTOR:
+        DISABLE_AUG_LIST: ['placeholder']
+        AUG_CONFIG_LIST:
             - NAME: gt_sampling
               USE_ROAD_PLANE: True
               DB_INFO_PATH: