From 4ad25cde30edf19f6bd128d6f54c6396cab49773 Mon Sep 17 00:00:00 2001 From: Logan Bauman Date: Wed, 4 May 2022 15:23:40 -0400 Subject: model arch started --- losses.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/losses.py b/losses.py index 0128749a..0affdb91 100644 --- a/losses.py +++ b/losses.py @@ -3,13 +3,18 @@ from tensorflow.keras.layers import \ Conv2D, MaxPool2D, Dropout, Flatten, Dense import numpy as np + +from skimage import transform import hyperparameters as hp class YourModel(tf.keras.Model): """ Your own neural network model. """ - def __init__(self): + def __init__(self, content_image, style_image): #normalize these images to float values super(YourModel, self).__init__() - + + self.content_image = content_image + self.style_image = transform.resize(style_image, np.shape(style_image), anti_aliasing=True) + self.x = tf.Variable(tf.random.uniform(np.shape(content_image)), trainable=True) self.alpha = 1 self.beta = 1 @@ -53,6 +58,8 @@ class YourModel(tf.keras.Model): activation="relu", name="block5_conv3"), MaxPool2D(2, name="block5_pool"), ] + for layer in self.vgg16: + layer.trainable = False self.head = [ # Dropout(.2), -- cgit v1.2.3-70-g09d2