diff --git a/README.md b/README.md index 6e34169..80b733e 100644 --- a/README.md +++ b/README.md @@ -33,6 +33,8 @@ python2.7 download.py -o python2.7 download.py -c bedroom # Download testing set python2.7 download.py -c test +# Download validation set +python2.7 download.py -c val ``` ## Demo code diff --git a/download.py b/download.py index 05090a1..22d13a5 100644 --- a/download.py +++ b/download.py @@ -50,6 +50,9 @@ def main(): else: if args.category == 'test': download(args.out_dir, '', 'test', args.tag) + elif args.category == 'val': + for category in categories: + download(args.out_dir, category, 'val', args.tag) elif args.category not in categories: print('Error:', args.category, "doesn't exist in", args.tag, 'LSUN release')