diff options
| -rw-r--r-- | hyperparameters.py | 4 | ||||
| -rw-r--r-- | losses.py | 2 | ||||
| -rw-r--r-- | 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 @@ -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 @@ -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) |