summaryrefslogtreecommitdiff
path: root/cli/app/search
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-23 03:00:16 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-23 03:00:16 +0100
commiteca62eb5244898eca3462404d266b184c8bbe091 (patch)
tree72a4301cafd962d76fe778384605c53e834ed120 /cli/app/search
parent036a0c608e4e43f5da3883c6324a7fca860b908b (diff)
more vgg
Diffstat (limited to 'cli/app/search')
-rw-r--r--cli/app/search/search_dense.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index 7b83952..503d135 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -495,7 +495,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
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_{}_{:04d}.png'.format(SAMPLES_DIR, opt_tag, int(it / 500)), inv_batch)
+ vs.save_image('{}/progress_{}_{:02d}_{:04d}.png'.format(SAMPLES_DIR, opt_tag, out_pos, int(it / 500)), inv_batch)
# Save images that are ready.
label_trained, latent_trained = sess.run([label, latent])