diff options
Diffstat (limited to 'cli')
| -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 c74c5f7..eee51d2 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -155,7 +155,7 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la # gen_img_ch = tf.transpose(gen_img / 2.0 + 0.5, [0, 2, 3, 1]) # target_img_ch = tf.transpose(target_img / 2.0 + 0.5, [0, 2, 3, 1]) - 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) # , axis=1) gen_img_ch = output # / 2.0 + 0.5 |