From 9608ec6a7bdf73d9d0d7fe406c575eb209cf50e0 Mon Sep 17 00:00:00 2001 From: Logan Bauman Date: Wed, 4 May 2022 15:05:30 -0400 Subject: finish total loss --- losses.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/losses.py b/losses.py index 6ebba671..caad3067 100644 --- a/losses.py +++ b/losses.py @@ -88,9 +88,14 @@ class YourModel(tf.keras.Model): _, art_layers = self.forward_pass(a) _, input_layers = self.forward_pass(x) + content_l = self.content_loss(photo_layers, input_layers) + style_l = self.style_loss(art_layers, input_layers) + # Equation 7 + return (self.alpha * content_l) + (self.beta * style_l) + - def content_loss(photo_layers, input_layers): + def content_loss(self, photo_layers, input_layers): L_content = tf.reduce_mean(tf.square(photo_layers - input_layers)) return L_content -- cgit v1.2.3-70-g09d2