Unverified Commit 22fdfde6 authored by VVsssssk's avatar VVsssssk Committed by GitHub
Browse files

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

* fix browse dataset and mmcv max version

* fix code
parent f747daab
...@@ -19,7 +19,7 @@ def digit_version(version_str): ...@@ -19,7 +19,7 @@ def digit_version(version_str):
mmcv_minimum_version = '1.4.8' mmcv_minimum_version = '1.4.8'
mmcv_maximum_version = '1.5.0' mmcv_maximum_version = '1.6.0'
mmcv_version = digit_version(mmcv.__version__) mmcv_version = digit_version(mmcv.__version__)
......
...@@ -79,6 +79,12 @@ def build_data_cfg(config_path, skip_type, aug, cfg_options): ...@@ -79,6 +79,12 @@ def build_data_cfg(config_path, skip_type, aug, cfg_options):
for i in range(len(cfg.train_pipeline)): for i in range(len(cfg.train_pipeline)):
if cfg.train_pipeline[i]['type'] == 'LoadAnnotations3D': if cfg.train_pipeline[i]['type'] == 'LoadAnnotations3D':
show_pipeline.insert(i, cfg.train_pipeline[i]) 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'] = [ train_data_cfg['pipeline'] = [
x for x in show_pipeline if x['type'] not in skip_type x for x in show_pipeline if x['type'] not in skip_type
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment