From 44f07d8dbbdace759150169cca0fafcaab8e1dd5 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 11 Dec 2019 03:31:10 +0100 Subject: hdf5 workaround --- inversion/image_inversion.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'inversion') 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) -- cgit v1.2.3-70-g09d2