diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 01:54:28 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 01:54:28 +0100 |
| commit | a194eaa66108d753aac1eac70b7016a9b20897e1 (patch) | |
| tree | 2e8678c9e0b5b3c6a1aa5587ad087c66dbfb85f0 /cli/app/search/search_class.py | |
| parent | daa10de26027528528f485b6dfed94864faf259a (diff) | |
up
Diffstat (limited to 'cli/app/search/search_class.py')
| -rw-r--r-- | cli/app/search/search_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index a4bec07..cd53a71 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -116,7 +116,7 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la ## if computing Feature loss, use these encoders if opt_use_feature_detector: print("Initializing feature detector...") - pix_square_diff = tf.square((target_img - gen_img) / 2.0) + pix_square_diff = tf.square((target - output) / 2.0) mse_loss = tf.reduce_mean(pix_square_diff) feature_extractor = hub.Module("https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1") |