diff options
| -rw-r--r-- | inversion/image_inversion.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 1201cf3..af8568f 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -433,7 +433,7 @@ for image_batch, label_batch in image_gen: if params.save_progress: assert SAMPLE_SIZE <= BATCH_SIZE gen_images = sess.run(gen_img) - inv_batch = vs.interleave(image_batch[BATCH_SIZE - SAMPLE_SIZE:], + inv_batch = vs.interleave(vs.data2img(image_batch[BATCH_SIZE - SAMPLE_SIZE:]), vs.data2img(gen_images[BATCH_SIZE - SAMPLE_SIZE:])) inv_batch = vs.grid_transform(inv_batch) vs.save_image('{}/progress_{}.png'.format(SAMPLES_DIR, it), inv_batch) @@ -445,7 +445,7 @@ for image_batch, label_batch in image_gen: custom_enc = gen_enc * (1-(j/10.0)) + enc_batch * (j/10.0) sess.run(encoding.assign(custom_enc)) gen_images = sess.run(gen_img) - inv_batch = vs.interleave(image_batch[BATCH_SIZE - SAMPLE_SIZE:], + inv_batch = vs.interleave(vs.data2img(image_batch[BATCH_SIZE - SAMPLE_SIZE:]), vs.data2img(gen_images[BATCH_SIZE - SAMPLE_SIZE:])) inv_batch = vs.grid_transform(inv_batch) vs.save_image('{}/progress_{}_lat_{}.png'.format(SAMPLES_DIR,it,j), @@ -459,7 +459,7 @@ for image_batch, label_batch in image_gen: if SAMPLE_SIZE > 0: assert SAMPLE_SIZE <= BATCH_SIZE gen_images = sess.run(gen_img) - inv_batch = vs.interleave(image_batch[BATCH_SIZE - SAMPLE_SIZE:], + inv_batch = vs.interleave(vs.data2img(image_batch[BATCH_SIZE - SAMPLE_SIZE:]), vs.data2img(gen_images[BATCH_SIZE - SAMPLE_SIZE:])) inv_batch = vs.grid_transform(inv_batch) vs.save_image('{}/{}.png'.format(SAMPLES_DIR, out_pos), inv_batch) |