summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-11 03:18:34 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-11 03:18:34 +0100
commit20a18eb423b90704752bf741258b0733b2f90403 (patch)
treeb9bc8744fca7f939c488c8b71eeabe367c6b1cb3
parentd0562102c6114bced962359303dd76d8fedf44a3 (diff)
what the...
-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 d91590c..04c99ab 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -194,7 +194,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
feat_loss_b, feat_err_b = feature_loss(feature_extractor, opt_feature_layers, BATCH_SIZE, gen_img_ch, target_img_ch, img_w - width, 0, height, width)
feat_loss_c, feat_err_c = feature_loss(feature_extractor, opt_feature_layers, BATCH_SIZE, gen_img_ch, target_img_ch, 0, img_w - width, height, width)
feat_loss_d, feat_err_d = feature_loss(feature_extractor, opt_feature_layers, BATCH_SIZE, gen_img_ch, target_img_ch, img_w - width, img_w - width, height, width)
- feat_loss_e, feat_err_e = feature_loss(feature_extractor, opt_feature_layers, BATCH_SIZE, gen_img_ch, target_img_ch, (img_w - width) / 2, (img_w - width) / 2, height, width)
+ feat_loss_e, feat_err_e = feature_loss(feature_extractor, opt_feature_layers, BATCH_SIZE, gen_img_ch, target_img_ch, int((img_w - width) / 2), int((img_w - width) / 2), height, width)
feat_loss_quad = feat_loss_a + feat_loss_b + feat_loss_c + feat_loss_d + feat_loss_e
img_feat_err_quad = feat_err_a + feat_err_b + feat_err_c + feat_err_d + feat_err_e