[Fix]fix browse dataset and mmcv max version (#1376)

* fix browse dataset and mmcv max version * fix code
open-mmlab · Apr 8, 2022 · 22fdfde · 22fdfde
1 parent f747daa
commit 22fdfde
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/mmdet3d/__init__.py b/mmdet3d/__init__.py
@@ -19,7 +19,7 @@ def digit_version(version_str):
 
 
 mmcv_minimum_version = '1.4.8'
-mmcv_maximum_version = '1.5.0'
+mmcv_maximum_version = '1.6.0'
 mmcv_version = digit_version(mmcv.__version__)
 
 

diff --git a/tools/misc/browse_dataset.py b/tools/misc/browse_dataset.py
@@ -79,6 +79,12 @@ def build_data_cfg(config_path, skip_type, aug, cfg_options):
         for i in range(len(cfg.train_pipeline)):
             if cfg.train_pipeline[i]['type'] == 'LoadAnnotations3D':
                 show_pipeline.insert(i, cfg.train_pipeline[i])
+            # Collect points as well as labels
+            if cfg.train_pipeline[i]['type'] == 'Collect3D':
+                if show_pipeline[-1]['type'] == 'Collect3D':
+                    show_pipeline[-1] = cfg.train_pipeline[i]
+                else:
+                    show_pipeline.append(cfg.train_pipeline[i])
 
     train_data_cfg['pipeline'] = [
         x for x in show_pipeline if x['type'] not in skip_type