diff options
Diffstat (limited to 'inversion/image_inversion.py')
| -rw-r--r-- | inversion/image_inversion.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 09dd4e7..43eb207 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -68,7 +68,7 @@ params = params.Params(sys.argv[1]) # -------------------------- # Global directories. # -------------------------- -LATENT_TAG = 'latent_' if params.inv_layer == 'latent' else 'dense_' +LATENT_TAG = 'latent' if params.inv_layer == 'latent' else 'dense' BATCH_SIZE = params.batch_size SAMPLE_SIZE = params.sample_size LOGS_DIR = os.path.join('inverses', params.tag, LATENT_TAG, 'logs') @@ -441,7 +441,7 @@ for image_batch, label_batch in image_gen: vs.save_image('{}/progress_{}_{}.png'.format(SAMPLES_DIR, params.tag, it), inv_batch) # Save linear interpolation between the actual and generated encodings. - if params.dist_loss and it % 1000 == 999: + if params.dist_loss and it % 1000 == 0: enc_batch, gen_enc = sess.run([encoding, gen_encoding]) for j in range(10): custom_enc = gen_enc * (1-(j/10.0)) + enc_batch * (j/10.0) |