Move use_valid_flag to nus (#106)

open-mmlab · Sep 16, 2020 · 9b09115 · 9b09115
1 parent 076b2cb
commit 9b09115
Showing 1 changed file with 16 additions and 17 deletions.
diff --git a/tools/data_converter/create_gt_database.py b/tools/data_converter/create_gt_database.py
@@ -142,10 +142,7 @@ def create_groundtruth_database(dataset_class_name,
     """
     print(f'Create GT Database of {dataset_class_name}')
     dataset_cfg = dict(
-        type=dataset_class_name,
-        data_root=data_path,
-        ann_file=info_path,
-        use_valid_flag=True)
+        type=dataset_class_name, data_root=data_path, ann_file=info_path)
     if dataset_class_name == 'KittiDataset':
         file_client_args = dict(backend='disk')
         dataset_cfg.update(
@@ -171,19 +168,21 @@ def create_groundtruth_database(dataset_class_name,
             ])
 
     elif dataset_class_name == 'NuScenesDataset':
-        dataset_cfg.update(pipeline=[
-            dict(type='LoadPointsFromFile', load_dim=5, use_dim=5),
-            dict(
-                type='LoadPointsFromMultiSweeps',
-                sweeps_num=10,
-                use_dim=[0, 1, 2, 3, 4],
-                pad_empty_sweeps=True,
-                remove_close=True),
-            dict(
-                type='LoadAnnotations3D',
-                with_bbox_3d=True,
-                with_label_3d=True)
-        ])
+        dataset_cfg.update(
+            use_valid_flag=True,
+            pipeline=[
+                dict(type='LoadPointsFromFile', load_dim=5, use_dim=5),
+                dict(
+                    type='LoadPointsFromMultiSweeps',
+                    sweeps_num=10,
+                    use_dim=[0, 1, 2, 3, 4],
+                    pad_empty_sweeps=True,
+                    remove_close=True),
+                dict(
+                    type='LoadAnnotations3D',
+                    with_bbox_3d=True,
+                    with_label_3d=True)
+            ])
     dataset = build_dataset(dataset_cfg)
 
     if database_save_path is None: