diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-02-20 23:27:55 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-02-20 23:27:55 +0100 |
| commit | b385b4ca946585effcdc90849b2ed9c840f29ad4 (patch) | |
| tree | 73c3b9d1623cbda0a9b7b9616a52e5775ab546fb /cli/app/search | |
| parent | a185742cecb7b2a7522ddb0b79a8eeb72e12a83d (diff) | |
vgg?
Diffstat (limited to 'cli/app/search')
| -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 |