From cea23978c222e71743c75b1ea6c31637166b62aa Mon Sep 17 00:00:00 2001 From: Florian Vahl Date: Fri, 18 Nov 2022 16:11:03 +0100 Subject: [PATCH] Remove combined zip --- scripts/download_dataset.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/scripts/download_dataset.py b/scripts/download_dataset.py index d36afba..4fb9862 100755 --- a/scripts/download_dataset.py +++ b/scripts/download_dataset.py @@ -48,11 +48,12 @@ def download_file(filename, file_url): parser.error('--all cannot be used with other options.') if args.all: - tmp_file = os.path.join(DATA_FOLDER, 'dataset.zip') - file_url = 'TORSO-21.zip' - folder = DATA_FOLDER - approx_size = 190 - download_and_extract_zip(tmp_file, file_url, folder, approx_size) + tmp_file = os.path.join(DATA_FOLDER, 'reality.zip') + approx_size = 11 + download_and_extract_zip(tmp_file, 'reality.zip', DATA_FOLDER, approx_size) + tmp_file = os.path.join(DATA_FOLDER, 'simulation.zip') + approx_size = 180 + download_and_extract_zip(tmp_file, 'simulation.zip', DATA_FOLDER, approx_size) elif args.real and args.test and args.annotations: download_file(os.path.join(DATA_FOLDER, 'reality', 'test', 'annotations.yaml'), 'reality/test/annotations.yaml') elif args.real and args.test: