diff options
Diffstat (limited to 'losses.py')
| -rw-r--r-- | losses.py | 6 |
1 files changed, 4 insertions, 2 deletions
@@ -34,7 +34,7 @@ class YourModel(tf.keras.Model): #print(self.content_image.shape, self.style_image.shape) - self.optimizer = tf.keras.optimizers.Adam() + self.optimizer = tf.keras.optimizers.Adam(1e-2) self.vgg16 = [ # Block 1 @@ -112,6 +112,8 @@ class YourModel(tf.keras.Model): content_l = self.content_loss(self.photo_layers, input_layers) style_l = self.style_loss(self.art_layers, input_layers) # Equation 7 + print('style_loss', style_l) + print('content_loss', content_l) return (self.alpha * content_l) + (self.beta * style_l) def content_loss(self, photo_layers, input_layers): @@ -164,7 +166,7 @@ class YourModel(tf.keras.Model): for i in range(len(art_layers)): art_layer = art_layers[i] input_layer = input_layers[i] - L_style = tf.math.add(L_style, self.layer_loss(art_layer, input_layer)) + L_style = tf.math.add(L_style, self.layer_loss(art_layer, input_layer)*(1/5)) #print('style loss', L_style) return L_style |