From 14f5c993ef8d0cce63f5fcd6cd4a88050ddaea8e Mon Sep 17 00:00:00 2001 From: Victor Arias <41467411+viariasv@users.noreply.github.com> Date: Thu, 7 Feb 2019 12:24:07 -0500 Subject: [PATCH] Update preprocessing.py more elegant image_specifiers, :). --- data/cityscapes/preprocessing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/cityscapes/preprocessing.py b/data/cityscapes/preprocessing.py index 1b17046..528f72f 100644 --- a/data/cityscapes/preprocessing.py +++ b/data/cityscapes/preprocessing.py @@ -63,7 +63,7 @@ def preprocess(cf): print('Processing {}'.format(city)) city_dir = os.path.join(image_dir, city) image_names = os.listdir(city_dir) - image_specifiers = ['_'.join(img.split('_')[:3]) for img in image_names] + image_specifiers = [img.rsplit('_', maxsplit=1)[0] for img in image_names] for img_spec in tqdm(image_specifiers): for scale in cf.settings[set]['resolutions']: @@ -96,4 +96,4 @@ def preprocess(cf): if __name__ == "__main__": cf = imp.load_source('cf', 'preprocessing_config.py') - preprocess(cf) \ No newline at end of file + preprocess(cf)