From b385b4ca946585effcdc90849b2ed9c840f29ad4 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 20 Feb 2020 23:27:55 +0100 Subject: vgg? --- cli/app/search/search_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli/app') 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 -- cgit v1.2.3-70-g09d2