From d7474266672176c8994858555bec8c4e40cbb662 Mon Sep 17 00:00:00 2001 From: Cameron Date: Wed, 16 Nov 2016 01:12:31 -0700 Subject: Update neural_style.py --- neural_style.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_style.py b/neural_style.py index 62875f6..8164b2c 100644 --- a/neural_style.py +++ b/neural_style.py @@ -479,9 +479,9 @@ def sum_total_variation_losses(sess, net, input_img): x = net['input'] tv_y_size = b * (h-1) * w * d tv_x_size = b * h * (w-1) * d - loss_y = tf.nn.l2_loss(x[:,1:,:,:] - x[:,:h-1,:,:]) + loss_y = tf.nn.l2_loss(x[:,1:,:,:] - x[:,:-1,:,:]) loss_y /= tv_y_size - loss_x = tf.nn.l2_loss(x[:,:,1:,:] - x[:,:,:w-1,:]) + loss_x = tf.nn.l2_loss(x[:,:,1:,:] - x[:,:,:-1,:]) loss_x /= tv_x_size loss = 2 * (loss_y + loss_x) loss = tf.cast(loss, tf.float32) -- cgit v1.2.3-70-g09d2