From 6a43ba28e63ab2db6e2b252a2f87863339f7fc74 Mon Sep 17 00:00:00 2001 From: Deep Patel Date: Fri, 29 Jun 2018 19:16:22 +0000 Subject: [PATCH 1/2] added ability to download validation set only --- README.md | 2 ++ download.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/README.md b/README.md index 6e34169..e7e8516 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 test ``` ## 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') From 4667209194b9b5126e421d3c64c44e6e00506bdc Mon Sep 17 00:00:00 2001 From: Deep Patel Date: Fri, 29 Jun 2018 19:18:36 +0000 Subject: [PATCH 2/2] fixed typo --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index e7e8516..80b733e 100644 --- a/README.md +++ b/README.md @@ -34,7 +34,7 @@ python2.7 download.py -c bedroom # Download testing set python2.7 download.py -c test # Download validation set -python2.7 download.py -c test +python2.7 download.py -c val ``` ## Demo code