Better sample plotting

openai · Dec 2, 2016 · 715a70c · 715a70c
1 parent f5294b1
commit 715a70c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/train_double_cnn.py b/train_double_cnn.py
@@ -247,10 +247,12 @@ def scale_x(x):
 
             # generate samples from the model
             sample_x = sample_from_model(sess)
-            img_tile = plotting.img_tile(sample_x, aspect_ratio=1.0, border_color=1.0, stretch=True)
-            img = plotting.plot_img(img_tile, title='CIFAR10 samples')
-            plotting.plt.savefig(args.save_dir + '/cifar10_sample' + str(epoch) + '.png')
-            plotting.plt.close('all')
+            #img_tile = plotting.img_tile(sample_x, aspect_ratio=1.0, border_color=1.0, stretch=True)
+            #img = plotting.plot_img(img_tile, title='CIFAR10 samples')
+            #plotting.plt.savefig(args.save_dir + '/cifar10_sample' + str(epoch) + '.png')
+            #plotting.plt.close('all')
+            import graphics
+            graphics.save_raster(sample_x, args.save_dir + '/cifar10_sample' + str(epoch) + '.png')
 
             # save params
             saver.save(sess, args.save_dir + '/params_' + args.data_set + '.ckpt')