From e5b07fb766660b5efddde822e95dbc884c6dd569 Mon Sep 17 00:00:00 2001
From: Piv <18462828+Piv200@users.noreply.github.com>
Date: Wed, 16 Jun 2021 21:52:31 +0930
Subject: [PATCH] Fix loss function

---
 losses.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/losses.py b/losses.py
index 55c3c3f..0e47ede 100644
--- a/losses.py
+++ b/losses.py
@@ -11,9 +11,10 @@ def dense_depth_loss_function(y, y_pred):
     # L1 loss over image gradients
     dy, dx = tf.image.image_gradients(y)
     dy_pred, dx_pred = tf.image.image_gradients(y_pred)
-    l_grad = tf.reduce_mean(tf.math.abs(dy_pred - dy) + tf.math.abs(dx_pred - dx), axis=-1)
+    l_grad = tf.reduce_mean(tf.math.abs(dy_pred - dy) +
+                            tf.math.abs(dx_pred - dx), axis=-1)
 
     #  Structural Similarity (SSIM)
     l_ssim = (1 - tf.image.ssim(y, y_pred, 500)) / 2
 
-    return 0.1 * tf.reduce_mean(l_depth) + l_grad + l_ssim
+    return 0.1 * tf.reduce_mean(l_depth) + tf.reduce_mean(l_grad) + l_ssim