Update prepare_data.py

a2321e7f · yizhou-wang · 476b4a19 · a2321e7f
Commit a2321e7f authored Nov 09, 2021 by yizhou-wang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tools/prepare_dataset/prepare_data.py tools/prepare_dataset/prepare_data.py +1 -0

No files found.
--- a/tools/prepare_dataset/prepare_data.py
+++ b/tools/prepare_dataset/prepare_data.py
@@ -207,6 +207,7 @@ if __name__ == "__main__":
    else:
        splits = args.split.split(',')
    out_data_dir = args.out_data_dir
+    os.makedirs(out_data_dir, exist_ok=True)
    overwrite = args.overwrite

    dataset = CRUW(data_root=data_root, sensor_config_name=args.sensor_config)