From 08a7caa870ea6a6ca92613e3133cceff460616c9 Mon Sep 17 00:00:00 2001 From: dpkingma Date: Thu, 1 Dec 2016 18:22:16 -0800 Subject: [PATCH] Changed location of cifar10_plotdata.py --- cifar10_plotdata.py => data/cifar10_plotdata.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) rename cifar10_plotdata.py => data/cifar10_plotdata.py (85%) diff --git a/cifar10_plotdata.py b/data/cifar10_plotdata.py similarity index 85% rename from cifar10_plotdata.py rename to data/cifar10_plotdata.py index 43e287623..179035bff 100644 --- a/cifar10_plotdata.py +++ b/data/cifar10_plotdata.py @@ -8,6 +8,7 @@ parser = argparse.ArgumentParser() parser.add_argument('--save_dir', type=str, default='./log') parser.add_argument('--data_dir', type=str, default='/home/tim/data') +parser.add_argument('--plot_title', type=str, default=None) args = parser.parse_args() print(args) @@ -29,7 +30,7 @@ print(ids) img_tile = plotting.img_tile(images, aspect_ratio=1.0, border_color=1.0, stretch=True) -img = plotting.plot_img(img_tile, title='samples from CIFAR10 dataset') +img = plotting.plot_img(img_tile, title=args.plot_title if args.plot_title != 'None' else None) plotting.plt.savefig(args.save_dir + '/cifar10_orig_images.png') plotting.plt.close('all')