From f094c415cb579af330c18dde68dd1596fc9064ef Mon Sep 17 00:00:00 2001 From: Steven Lang Date: Wed, 2 Jan 2019 21:51:43 +0100 Subject: [PATCH] Fix disparity smoothness missing y gradient --- monodepth_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monodepth_model.py b/monodepth_model.py index b66b684..37ec061 100644 --- a/monodepth_model.py +++ b/monodepth_model.py @@ -352,8 +352,8 @@ def build_losses(self): self.image_loss = tf.add_n(self.image_loss_left + self.image_loss_right) # DISPARITY SMOOTHNESS - self.disp_left_loss = [tf.reduce_mean(tf.abs(self.disp_left_smoothness[i])) / 2 ** i for i in range(4)] - self.disp_right_loss = [tf.reduce_mean(tf.abs(self.disp_right_smoothness[i])) / 2 ** i for i in range(4)] + self.disp_left_loss = [tf.reduce_mean(tf.abs(self.disp_left_smoothness[i])) / 2 ** i for i in range(4 * 2)] + self.disp_right_loss = [tf.reduce_mean(tf.abs(self.disp_right_smoothness[i])) / 2 ** i for i in range(4 * 2)] self.disp_gradient_loss = tf.add_n(self.disp_left_loss + self.disp_right_loss) # LR CONSISTENCY