summaryrefslogtreecommitdiff
path: root/cli/app/search/search_dense.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-16 12:56:05 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-16 12:56:05 +0100
commit288b27c004959ae25bb637072e945a5b07b5abe5 (patch)
treef7ea9ac0982cc99a87d25342056de34b4479163c /cli/app/search/search_dense.py
parent36d6ab100116f60deb33b622910b1132cb1752e1 (diff)
ugh
Diffstat (limited to 'cli/app/search/search_dense.py')
-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 a07e15b..96ed4af 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -441,8 +441,8 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
if out_i >= NUM_IMGS:
print("{} >= {}, skipping...".format(out_i, NUM_IMGS))
continue
- print("{}: {}".format(out_i, sample_fn))
sample_fn, ext = os.path.splitext(sample_fns[out_i])
+ print("{}: {}".format(out_i, sample_fn))
image = Image.fromarray(images[i])
fp = BytesIO()
image.save(fp, format='png')