diff --git a/downloader.py b/downloader.py index 5bf3c2c0..111c1f98 100755 --- a/downloader.py +++ b/downloader.py @@ -33,8 +33,8 @@ exit() if not os.path.isdir(args.data_root): - logging.error(f'folder {args.data_root} does not exist! please provide existing folder in -data_root arg!') - exit() + logging.error(f'folder {args.data_root} does not exist! Creating new folder') + os.makedirs(os.path.join(os.getcwd(),args.data_root)) IMAGENET_API_WNID_TO_URLS = lambda wnid: f'http://www.image-net.org/api/text/imagenet.synset.geturls?wnid={wnid}'