From 83e91e0a8effcd20466e56b6ecc3e349bbfa5e0e Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Tue, 24 Apr 2018 20:19:24 +0200 Subject: updates to tensorflow code and processing additions --- Code/loss_functions.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'Code/loss_functions.py') diff --git a/Code/loss_functions.py b/Code/loss_functions.py index 994d226..0d33b49 100644 --- a/Code/loss_functions.py +++ b/Code/loss_functions.py @@ -61,7 +61,7 @@ def lp_loss(gen_frames, gt_frames, l_num): scale_losses.append(tf.reduce_sum(tf.abs(gen_frames[i] - gt_frames[i])**l_num)) # condense into one tensor and avg - return tf.reduce_mean(tf.pack(scale_losses)) + return tf.reduce_mean(tf.stack(scale_losses)) def gdl_loss(gen_frames, gt_frames, alpha): @@ -80,8 +80,8 @@ def gdl_loss(gen_frames, gt_frames, alpha): # create filters [-1, 1] and [[1],[-1]] for diffing to the left and down respectively. pos = tf.constant(np.identity(3), dtype=tf.float32) neg = -1 * pos - filter_x = tf.expand_dims(tf.pack([neg, pos]), 0) # [-1, 1] - filter_y = tf.pack([tf.expand_dims(pos, 0), tf.expand_dims(neg, 0)]) # [[1],[-1]] + filter_x = tf.expand_dims(tf.stack([neg, pos]), 0) # [-1, 1] + filter_y = tf.stack([tf.expand_dims(pos, 0), tf.expand_dims(neg, 0)]) # [[1],[-1]] strides = [1, 1, 1, 1] # stride of (1, 1) padding = 'SAME' @@ -96,7 +96,7 @@ def gdl_loss(gen_frames, gt_frames, alpha): scale_losses.append(tf.reduce_sum((grad_diff_x ** alpha + grad_diff_y ** alpha))) # condense into one tensor and avg - return tf.reduce_mean(tf.pack(scale_losses)) + return tf.reduce_mean(tf.stack(scale_losses)) def adv_loss(preds, labels): @@ -115,4 +115,4 @@ def adv_loss(preds, labels): scale_losses.append(loss) # condense into one tensor and avg - return tf.reduce_mean(tf.pack(scale_losses)) + return tf.reduce_mean(tf.stack(scale_losses)) -- cgit v1.2.3-70-g09d2