From 46a7929942f90d960a0a4d6e35251c900bd45fb2 Mon Sep 17 00:00:00 2001
From: Logan Bauman <logan_bauman@brown.edu>
Date: Sat, 7 May 2022 08:15:54 -0400
Subject: patch preprocessing, fix layer loss and hyper params

---
 hyperparameters.py | 4 ++--
 losses.py          | 2 +-
 main.py            | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/hyperparameters.py b/hyperparameters.py
index a0068dd1..63e51b91 100644
--- a/hyperparameters.py
+++ b/hyperparameters.py
@@ -20,6 +20,6 @@ learning_rate = 3e-2
 
 momentum = 0.01
 
-alpha = 1e-2
+alpha = 1e-5
 
-beta = 1e-5
+beta = 1e-2
diff --git a/losses.py b/losses.py
index 407412a1..8c142f86 100644
--- a/losses.py
+++ b/losses.py
@@ -145,7 +145,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, (1/5)*self.layer_loss(art_layer, input_layer))
         #print('style loss', L_style)
         return L_style
             
diff --git a/main.py b/main.py
index de769baa..b7d9a1e0 100644
--- a/main.py
+++ b/main.py
@@ -11,6 +11,7 @@ from losses import YourModel
 from skimage.io import imread, imsave
 from matplotlib import pyplot as plt
 import numpy as np
+from skimage import transform
 
 
 def parse_args():
@@ -49,9 +50,8 @@ def main():
     print('this is',ARGS.content)
 
     content_image = imread(ARGS.content)
-    content_image = np.resize(content_image, (255, 255, 3))
     style_image = imread(ARGS.style)
-    style_image = np.resize(style_image, (255, 255, 3))
+    style_image = transform.resize(style_image, content_image.shape)
     my_model = YourModel(content_image=content_image, style_image=style_image)
     my_model.vgg16.build([1, 255, 255, 3])
     my_model.vgg16.load_weights('vgg16_imagenet.h5', by_name=True)
-- 
cgit v1.2.3-70-g09d2