diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 03:31:10 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 03:31:10 +0100 |
| commit | 44f07d8dbbdace759150169cca0fafcaab8e1dd5 (patch) | |
| tree | a457d6296a1422f5e1bc9349bad8f69fecf62c1b /inversion | |
| parent | a982f0ce441cf2321b7bcae63578db4dea20690d (diff) | |
hdf5 workaround
Diffstat (limited to 'inversion')
| -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) |