Merge branch 'master' of asdf.us:pix2pixhd

1 files changed, 3 insertions, 1 deletions

diff --git a/augment.py b/augment.py
index e1360b0..c0f8c7a 100755
--- a/augment.py
+++ b/augment.py
@@ -122,6 +122,8 @@ for m in range(data_opt.augment_take):
   i = randint(0, _len)
   index = i
 
+  print('take frame... %i' % index)
+
   for n in range(data_opt.augment_make):
     index = i + n
     if n == 0:
@@ -154,7 +156,7 @@ for m in range(data_opt.augment_take):
     
     tmp_path = os.path.join(opt.render_dir, "recur_{}_{:05d}_{:05d}_tmp.png".format(epoch_id, m, index+1))
     next_path = os.path.join(opt.render_dir, "recur_{}_{:05d}_{:05d}.png".format(epoch_id, m, index+1))
-    print('process image... %i' % index)
+    # print('process image... %i' % index)
 
     im = util.tensor2im(generated.data[0])
     image_pil = Image.fromarray(im, mode='RGB')