Starting from:
$30
Deep Learning & Art: Neural Style Transfer
Deep Learning & Art: Neural Style Transfer
Welcome to the second assignment of this week. In this assignment, you will learn about Neural Style Transfer. This algorithm was created by Gatys et al. (2015) (https://arxiv.org/abs/1508.06576 (https://arxiv.org /abs/1508.06576)). In this assignment, you will: Implement the neural style transfer algorithm Generate novel artistic images using your algorithm Most of the algorithms you've studied optimize a cost function to get a set of parameter values. In Neural Style Transfer, you'll optimize a cost function to get pixel values! In [1]: import os import sys import scipy.io import scipy.misc import matplotlib.pyplot as plt from matplotlib.pyplot import imshow from PIL import Image from nst_utils import * import numpy as np import tensorflow as tf %matplotlib inline 1 - Problem Statement Neural Style Transfer (NST) is one of the most fun techniques in deep learning. As seen below, it merges two images, namely, a "content" image (C) and a "style" image (S), to create a "generated" image (G). The generated image G combines the "content" of the image C with the "style" of image S. In this example, you are going to generate an image of the Louvre museum in Paris (content image C), mixed with a painting by Claude Monet, a leader of the impressionist movement (style image S). Let's see how you can do this. 2 - Transfer Learning Neural Style Transfer (NST) uses a previously trained convolutional network, and builds on top of that. The idea of using a network trained on a different task and applying it to a new task is called transfer learning. Following the original NST paper (https://arxiv.org/abs/1508.06576 (https://arxiv.org/abs/1508.06576)), we will use the VGG network. Specifically, we'll use VGG-19, a 19-layer version of the VGG network. This model has already been trained on the very large ImageNet database, and thus has learned to recognize a variety of low level features (at the earlier layers) and high level features (at the deeper layers). Run the following code to load parameters from the VGG model. This may take a few seconds. In [2]: model = load_vgg_model("pretrained-model/imagenet-vgg-verydeep-19.mat") print(model) The model is stored in a python dictionary where each variable name is the key and the corresponding value is a tensor containing that variable's value. To run an image through this network, you just have to feed the image to the model. In TensorFlow, you can do so using the tf.assign (https://www.tensorflow.org/api_docs /python/tf/assign) function. In particular, you will use the assign function like this: model["input"].assign(image) This assigns the image as an input to the model. After this, if you want to access the activations of a particular layer, say layer 4_2 when the network is run on this image, you would run a TensorFlow session on the correct tensor conv4_2, as follows: sess.run(model["conv4_2"]) {'input': <tf.Variable 'Variable:0' shape=(1, 300, 400, 3) dtype=floa t32_ref, 'conv1_1': <tf.Tensor 'Relu:0' shape=(1, 300, 400, 64) dtyp e=float32, 'conv1_2': <tf.Tensor 'Relu_1:0' shape=(1, 300, 400, 64) dtype=float32, 'avgpool1': <tf.Tensor 'AvgPool:0' shape=(1, 150, 200 , 64) dtype=float32, 'conv2_1': <tf.Tensor 'Relu_2:0' shape=(1, 150, 200, 128) dtype=float32, 'conv2_2': <tf.Tensor 'Relu_3:0' shape=(1, 150, 200, 128) dtype=float32, 'avgpool2': <tf.Tensor 'AvgPool_1:0' s hape=(1, 75, 100, 128) dtype=float32, 'conv3_1': <tf.Tensor 'Relu_4: 0' shape=(1, 75, 100, 256) dtype=float32, 'conv3_2': <tf.Tensor 'Rel u_5:0' shape=(1, 75, 100, 256) dtype=float32, 'conv3_3': , 'conv3_4': , 'avgpool3': , 'conv4_ 1': <tf.Tensor 'Relu_8:0' shape=(1, 38, 50, 512) dtype=float32, 'con v4_2': <tf.Tensor 'Relu_9:0' shape=(1, 38, 50, 512) dtype=float32, ' conv4_3': <tf.Tensor 'Relu_10:0' shape=(1, 38, 50, 512) dtype=float32 , 'conv4_4': <tf.Tensor 'Relu_11:0' shape=(1, 38, 50, 512) dtype=flo at32, 'avgpool4': <tf.Tensor 'AvgPool_3:0' shape=(1, 19, 25, 512) dt ype=float32, 'conv5_1': <tf.Tensor 'Relu_12:0' shape=(1, 19, 25, 512 ) dtype=float32, 'conv5_2': <tf.Tensor 'Relu_13:0' shape=(1, 19, 25, 512) dtype=float32, 'conv5_3': <tf.Tensor 'Relu_14:0' shape=(1, 19, 25, 512) dtype=float32, 'conv5_4': <tf.Tensor 'Relu_15:0' shape=(1, 19, 25, 512) dtype=float32, 'avgpool5': <tf.Tensor 'AvgPool_4:0' sha pe=(1, 10, 13, 512) dtype=float32} 3 - Neural Style Transfer We will build the NST algorithm in three steps: Build the content cost function Build the style cost function Put it together to get . 3.1 - Computing the content cost In our running example, the content image C will be the picture of the Louvre Museum in Paris. Run the code below to see a picture of the Louvre. In [3]: content_image = scipy.misc.imread("images/louvre.jpg") imshow(content_image) Out[3]: The content image (C) shows the Louvre museum's pyramid surrounded by old Paris buildings, against a sunny sky with a few clouds. 3.1.1 - How do you ensure the generated image G matches the content of the image C? As we saw in lecture, the earlier (shallower) layers of a ConvNet tend to detect lower-level features such as edges and simple textures, and the later (deeper) layers tend to detect higher-level features such as more complex textures as well as object classes. We would like the "generated" image G to have similar content as the input image C. Suppose you have chosen some layer's activations to represent the content of an image. In practice, you'll get the most visually pleasing results if you choose a layer in the middle of the network--neither too shallow nor too deep. (After you have finished this exercise, feel free to come back and experiment with using different layers, to see how the results vary.) So, suppose you have picked one particular hidden layer to use. Now, set the image C as the input to the pretrained VGG network, and run forward propagation. Let be the hidden layer activations in the layer you had chosen. (In lecture, we had written this as , but here we'll drop the superscript to simplify the notation.) This will be a tensor. Repeat this process with the image G: Set G as the input, and run forward progation. Let be the corresponding hidden layer activation. We will define as the content cost function as: Here, and are the height, width and number of channels of the hidden layer you have chosen, and appear in a normalization term in the cost. For clarity, note that and are the volumes corresponding to a hidden layer's activations. In order to compute the cost , it might also be convenient to unroll these 3D volumes into a 2D matrix, as shown below. (Technically this unrolling step isn't needed to compute , but it will be good practice for when you do need to carry out a similar operation later for computing the style const .) In [4]: # GRADED FUNCTION: compute_content_cost def compute_content_cost(a_C, a_G): """ Computes the content cost Arguments: a_C -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing content of the image C a_G -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing content of the image G Returns: J_content -- scalar that you compute using equation 1 above. """ ### START CODE HERE ### # Retrieve dimensions from a_G (≈1 line) m, n_H, n_W, n_C = a_G.get_shape().as_list() # Reshape a_C and a_G (≈2 lines) a_C_unrolled = tf.reshape(a_C, [n_H * n_W, n_C]) a_G_unrolled = tf.reshape(a_G, [n_H * n_W, n_C]) # compute the cost with tensorflow (≈1 line) J_content = tf.reduce_sum(tf.square(tf.subtract(a_C_unrolled, a_G_ unrolled))) / (4 * n_H * n_W * n_C) ### END CODE HERE ### return J_content In [5]: tf.reset_default_graph() with tf.Session() as test: tf.set_random_seed(1) a_C = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) a_G = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) J_content = compute_content_cost(a_C, a_G) print("J_content = " + str(J_content.eval())) Expected Output: **J_content** 6.76559 What you should remember: The content cost takes a hidden layer activation of the neural network, and measures how different and are. When we minimize the content cost later, this will help make sure has similar content as . J_content = 6.76559 3.2 - Computing the style cost For our running example, we will use the following style image: In [6]: style_image = scipy.misc.imread("images/monet_800600.jpg") imshow(style_image) This painting was painted in the style of impressionism (https://en.wikipedia.org/wiki/Impressionism). Lets see how you can now define a "style" const function . Out[6]: 3.2.1 - Style matrix The style matrix is also called a "Gram matrix." In linear algebra, the Gram matrix G of a set of vectors is the matrix of dot products, whose entries are . In other words, compares how similar is to : If they are highly similar, you would expect them to have a large dot product, and thus for to be large. Note that there is an unfortunate collision in the variable names used here. We are following common terminology used in the literature, but is used to denote the Style matrix (or Gram matrix) as well as to denote the generated image . We will try to make sure which we are referring to is always clear from the context. In NST, you can compute the Style matrix by multiplying the "unrolled" filter matrix with their transpose: The result is a matrix of dimension where is the number of filters. The value measures how similar the activations of filter are to the activations of filter . One important part of the gram matrix is that the diagonal elements such as also measures how active filter is. For example, suppose filter is detecting vertical textures in the image. Then measures how common vertical textures are in the image as a whole: If is large, this means that the image has a lot of vertical texture. By capturing the prevalence of different types of features ( ), as well as how much different features occur together ( ), the Style matrix measures the style of an image. Exercise: Using TensorFlow, implement a function that computes the Gram matrix of a matrix A. The formula is: The gram matrix of A is . If you are stuck, take a look at Hint 1 (https://www.tensorflow.org /api_docs/python/tf/matmul) and Hint 2 (https://www.tensorflow.org/api_docs/python/tf/transpose). In [7]: # GRADED FUNCTION: gram_matrix def gram_matrix(A): """ Argument: A -- matrix of shape (n_C, n_H*n_W) Returns: GA -- Gram matrix of A, of shape (n_C, n_C) """ ### START CODE HERE ### (≈1 line) GA = tf.matmul(A, A, transpose_a=False, transpose_b=True,) ### END CODE HERE ### return GA In [8]: tf.reset_default_graph() with tf.Session() as test: tf.set_random_seed(1) A = tf.random_normal([3, 2*1], mean=1, stddev=4) GA = gram_matrix(A) print("GA = " + str(GA.eval())) Expected Output: **GA** [[ 6.42230511 -4.42912197 -2.09668207] [ -4.42912197 19.46583748 19.56387138] [ -2.09668207 19.56387138 20.6864624 ]] 3.2.2 - Style cost After generating the Style matrix (Gram matrix), your goal will be to minimize the distance between the Gram matrix of the "style" image S and that of the "generated" image G. For now, we are using only a single hidden layer , and the corresponding style cost for this layer is defined as: where and are respectively the Gram matrices of the "style" image and the "generated" image, computed using the hidden layer activations for a particular hidden layer in the network. GA = [[ 6.42230511 -4.42912197 -2.09668207] [ -4.42912197 19.46583748 19.56387138] [ -2.09668207 19.56387138 20.6864624 ]] Exercise: Compute the style cost for a single layer. Instructions: The 3 steps to implement this function are: Retrieve dimensions from the hidden layer activations a_G: To retrieve dimensions from a tensor X, use: X.get_shape().as_list() 1. Unroll the hidden layer activations a_S and a_G into 2D matrices, as explained in the picture above. You may find Hint1 (https://www.tensorflow.org/versions/r1.3/api_docs/python/tf/transpose) and Hint2 (https://www.tensorflow.org/versions/r1.2/api_docs/python/tf/reshape) useful. 2. 3. Compute the Style matrix of the images S and G. (Use the function you had previously written.) Compute the Style cost: You may find Hint3 (https://www.tensorflow.org/api_docs/python/tf/reduce_sum), Hint4 (https://www.tensorflow.org/api_docs/python/tf/square) and Hint5 (https://www.tensorflow.org/api_docs/python/tf/subtract) useful. 4. In [9]: # GRADED FUNCTION: compute_layer_style_cost def compute_layer_style_cost(a_S, a_G): """ Arguments: a_S -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing style of the image S a_G -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing style of the image G Returns: J_style_layer -- tensor representing a scalar value, style cost def ined above by equation (2) """ ### START CODE HERE ### # Retrieve dimensions from a_G (≈1 line) m, n_H, n_W, n_C = a_G.get_shape().as_list() # Reshape the images to have them of shape (n_C, n_H*n_W) (≈2 lines ) a_S = tf.transpose(tf.reshape(a_S, [n_H * n_W, n_C])) a_G = tf.transpose(tf.reshape(a_G, [n_H * n_W, n_C])) # Computing gram_matrices for both images S and G (≈2 lines) GS = gram_matrix(a_S) GG = gram_matrix(a_G) # Computing the loss (≈1 line) J_style_layer = tf.reduce_sum(tf.square(tf.subtract(GS, GG))) / (4 * np.square(n_C) * np.square(n_W * n_H) ) ### END CODE HERE ### return J_style_layer In [10]: tf.reset_default_graph() with tf.Session() as test: tf.set_random_seed(1) a_S = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) a_G = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) J_style_layer = compute_layer_style_cost(a_S, a_G) print("J_style_layer = " + str(J_style_layer.eval())) Expected Output: **J_style_layer** 9.19028 3.2.3 Style Weights So far you have captured the style from only one layer. We'll get better results if we "merge" style costs from several different layers. After completing this exercise, feel free to come back and experiment with different weights to see how it changes the generated image . But for now, this is a pretty reasonable default: In [11]: STYLE_LAYERS = [ ('conv1_1', 0.2), ('conv2_1', 0.2), ('conv3_1', 0.2), ('conv4_1', 0.2), ('conv5_1', 0.2)] You can combine the style costs for different layers as follows: where the values for are given in STYLE_LAYERS. We've implemented a compute_style_cost(...) function. It simply calls your compute_layer_style_cost(...) several times, and weights their results using the values in STYLE_LAYERS. Read over it to make sure you understand what it's doing. J_style_layer = 9.19028 In [12]: def compute_style_cost(model, STYLE_LAYERS): """ Computes the overall style cost from several chosen layers Arguments: model -- our tensorflow model STYLE_LAYERS -- A python list containing: - the names of the layers we would like to extr act style from - a coefficient for each of them Returns: J_style -- tensor representing a scalar value, style cost defined a bove by equation (2) """ # initialize the overall style cost J_style = 0 for layer_name, coeff in STYLE_LAYERS: # Select the output tensor of the currently selected layer out = model[layer_name] # Set a_S to be the hidden layer activation from the layer we h ave selected, by running the session on out a_S = sess.run(out) # Set a_G to be the hidden layer activation from same layer. He re, a_G references model[layer_name] # and isn't evaluated yet. Later in the code, we'll assign the image G as the model input, so that # when we run the session, this will be the activations drawn f rom the appropriate layer, with G as input. a_G = out # Compute style_cost for the current layer J_style_layer = compute_layer_style_cost(a_S, a_G) # Add coeff * J_style_layer of this layer to overall style cost J_style += coeff * J_style_layer return J_style Note: In the inner-loop of the for-loop above, a_G is a tensor and hasn't been evaluated yet. It will be evaluated and updated at each iteration when we run the TensorFlow graph in model_nn() below. What you should remember: The style of an image can be represented using the Gram matrix of a hidden layer's activations. However, we get even better results combining this representation from multiple different layers. This is in contrast to the content representation, where usually using just a single hidden layer is sufficient. Minimizing the style cost will cause the image to follow the style of the image . </font color='blue' 3.3 - Defining the total cost to optimize Finally, let's create a cost function that minimizes both the style and the content cost. The formula is: Exercise: Implement the total cost function which includes both the content cost and the style cost. In [13]: # GRADED FUNCTION: total_cost def total_cost(J_content, J_style, alpha = 10, beta = 40): """ Computes the total cost function Arguments: J_content -- content cost coded above J_style -- style cost coded above alpha -- hyperparameter weighting the importance of the content cos t beta -- hyperparameter weighting the importance of the style cost Returns: J -- total cost as defined by the formula above. """ ### START CODE HERE ### (≈1 line) J = (alpha * J_content) + (beta * J_style) ### END CODE HERE ### return J In [14]: tf.reset_default_graph() with tf.Session() as test: np.random.seed(3) J_content = np.random.randn() J_style = np.random.randn() J = total_cost(J_content, J_style) print("J = " + str(J)) Expected Output: **J** 35.34667875478276 What you should remember: The total cost is a linear combination of the content cost and the style cost and are hyperparameters that control the relative weighting between content and style 4 - Solving the optimization problem Finally, let's put everything together to implement Neural Style Transfer! Here's what the program will have to do: 1. Create an Interactive Session 2. Load the content image 3. Load the style image 4. Randomly initialize the image to be generated 5. Load the VGG16 model Build the TensorFlow graph: Run the content image through the VGG16 model and compute the content cost Run the style image through the VGG16 model and compute the style cost Compute the total cost Define the optimizer and the learning rate 6. Initialize the TensorFlow graph and run it for a large number of iterations, updating the generated image at every step. 7. Lets go through the individual steps in detail. J = 35.34667875478276 You've previously implemented the overall cost . We'll now set up TensorFlow to optimize this with respect to . To do so, your program has to reset the graph and use an "Interactive Session (https://www.tensorflow.org/api_docs/python/tf/InteractiveSession)". Unlike a regular session, the "Interactive Session" installs itself as the default session to build a graph. This allows you to run variables without constantly needing to refer to the session object, which simplifies the code. Lets start the interactive session. In [15]: # Reset the graph tf.reset_default_graph() # Start interactive session sess = tf.InteractiveSession() Let's load, reshape, and normalize our "content" image (the Louvre museum picture): In [16]: content_image = scipy.misc.imread("images/louvre_small.jpg") content_image = reshape_and_normalize_image(content_image) Let's load, reshape and normalize our "style" image (Claude Monet's painting): In [17]: style_image = scipy.misc.imread("images/monet.jpg") style_image = reshape_and_normalize_image(style_image) Now, we initialize the "generated" image as a noisy image created from the content_image. By initializing the pixels of the generated image to be mostly noise but still slightly correlated with the content image, this will help the content of the "generated" image more rapidly match the content of the "content" image. (Feel free to look in nst_utils.py to see the details of generate_noise_image(...); to do so, click "File--Open..." at the upper-left corner of this Jupyter notebook.) In [18]: generated_image = generate_noise_image(content_image) imshow(generated_image[0]) Next, as explained in part (2), let's load the VGG16 model. In [19]: model = load_vgg_model("pretrained-model/imagenet-vgg-verydeep-19.mat") To get the program to compute the content cost, we will now assign a_C and a_G to be the appropriate hidden layer activations. We will use layer conv4_2 to compute the content cost. The code below does the following: 1. Assign the content image to be the input to the VGG model. 2. Set a_C to be the tensor giving the hidden layer activation for layer "conv4_2". 3. Set a_G to be the tensor giving the hidden layer activation for the same layer. 4. Compute the content cost using a_C and a_G. Out[18]: In [20]: # Assign the content image to be the input of the VGG model. sess.run(model['input'].assign(content_image)) # Select the output tensor of layer conv4_2 out = model['conv4_2'] # Set a_C to be the hidden layer activation from the layer we have sele cted a_C = sess.run(out) # Set a_G to be the hidden layer activation from same layer. Here, a_G references model['conv4_2'] # and isn't evaluated yet. Later in the code, we'll assign the image G as the model input, so that # when we run the session, this will be the activations drawn from the appropriate layer, with G as input. a_G = out # Compute the content cost J_content = compute_content_cost(a_C, a_G) Note: At this point, a_G is a tensor and hasn't been evaluated. It will be evaluated and updated at each iteration when we run the Tensorflow graph in model_nn() below. In [21]: # Assign the input of the model to be the "style" image sess.run(model['input'].assign(style_image)) # Compute the style cost J_style = compute_style_cost(model, STYLE_LAYERS) Exercise: Now that you have J_content and J_style, compute the total cost J by calling total_cost(). Use alpha = 10 and beta = 40. In [22]: ### START CODE HERE ### (1 line) J = total_cost(J_content, J_style, alpha = 10, beta = 40) ### END CODE HERE ### You'd previously learned how to set up the Adam optimizer in TensorFlow. Lets do that here, using a learning rate of 2.0. See reference (https://www.tensorflow.org/api_docs/python/tf/train/AdamOptimizer) In [23]: # define optimizer (1 line) optimizer = tf.train.AdamOptimizer(2.0) # define train_step (1 line) train_step = optimizer.minimize(J) Exercise: Implement the model_nn() function which initializes the variables of the tensorflow graph, assigns the input image (initial generated image) as the input of the VGG16 model and runs the train_step for a large number of steps. In [32]: def model_nn(sess, input_image, num_iterations = 200): # Initialize global variables (you need to run the session on the i nitializer) ### START CODE HERE ### (1 line) sess.run(tf.global_variables_initializer()) ### END CODE HERE ### # Run the noisy input image (initial generated image) through the m odel. Use assign(). ### START CODE HERE ### (1 line) sess.run(model['input'].assign(input_image)) ### END CODE HERE ### for i in range(num_iterations): # Run the session on the train_step to minimize the total cost ### START CODE HERE ### (1 line) sess.run(train_step) ### END CODE HERE ### # Compute the generated image by running the session on the cur rent model['input'] ### START CODE HERE ### (1 line) generated_image = sess.run(model['input']) ### END CODE HERE ### # Print every 20 iteration. if i%20 == 0: Jt, Jc, Js = sess.run([J, J_content, J_style]) print("Iteration " + str(i) + " :") print("total cost = " + str(Jt)) print("content cost = " + str(Jc)) print("style cost = " + str(Js)) # save current generated image in the "/output" directory save_image("output/" + str(i) + ".png", generated_image) # save last generated image save_image('output/generated_image.jpg', generated_image) return generated_image Run the following cell to generate an artistic image. It should take about 3min on CPU for every 20 iterations but you start observing attractive results after ≈140 iterations. Neural Style Transfer is generally trained using GPUs. In [33]: model_nn(sess, generated_image) Iteration 0 : total cost = 5.05035e+09 content cost = 7877.67 style cost = 1.26257e+08 Iteration 20 : total cost = 9.43282e+08 content cost = 15187.1 style cost = 2.35783e+07 Iteration 40 : total cost = 4.84914e+08 content cost = 16785.8 style cost = 1.21187e+07 Iteration 60 : total cost = 3.12578e+08 content cost = 17465.4 style cost = 7.81007e+06 Iteration 80 : total cost = 2.28117e+08 content cost = 17715.0 style cost = 5.69849e+06 Iteration 100 : total cost = 1.80678e+08 content cost = 17897.0 style cost = 4.51247e+06 Iteration 120 : total cost = 1.49979e+08 content cost = 18036.8 style cost = 3.74497e+06 Iteration 140 : total cost = 1.27725e+08 content cost = 18192.9 style cost = 3.18859e+06 Iteration 160 : total cost = 1.107e+08 content cost = 18353.7 style cost = 2.76292e+06 Iteration 180 : total cost = 9.73286e+07 content cost = 18498.0 style cost = 2.42859e+06 Out[33]: array([[[[ -47.79544449, -61.47901154, 48.77702713], [ -26.42229462, -40.57945633, 26.92292023], [ -41.86124039, -28.98380089, 11.30953312], ..., [ -26.89013481, -9.54228497, 14.28584766], [ -30.29121208, -2.94760489, 24.04436874], [ -42.41747665, -4.17948103, 49.22496033]], [[ -61.32366562, -51.78530502, 25.22885132], [ -33.15329361, -31.11307716, -1.55355597], [ -27.22502708, -30.70824814, 15.15648651], ..., [ -26.75429535, -4.98880053, 26.1478672 ], [ -21.36663246, -16.82632828, 14.10803032], [ -40.67829895, -6.07156563, 9.60023594]], [[ -52.3535614 , -51.64604187, 13.5016737 ], [ -37.27353668, -41.49959183, -6.35776997], [ -34.27314377, -25.33574867, 7.28916264], ..., [ -10.58620262, -37.16160965, 12.67485428], [ -12.21832657, -20.8809433 , 17.26901627], [ -22.42559814, -18.40380859, 14.41193676]], ..., [[ -49.01599503, -53.96035004, -37.23251724], [ -97.29430389, -76.19354248, -267.10494995], [ -74.4513092 , -70.41262054, -141.00666809], ..., [ -69.40086365, -70.02838898, -29.45741272], [ -78.91752625, -87.95886993, -22.85899162], [ 1.8733685 , -39.92020798, 23.59747505]], [[ -1.7835474 , -74.61296082, 13.84414291], [-172.76348877, -101.91304016, -30.54630089], [ 7.31526756, -70.08149719, -20.32836342], ..., [ -95.6863327 , -84.15804291, -47.7417717 ], [-102.43443298, -102.8863678 , -59.78057098], [ -65.53679657, -95.97616577, 1.40922606]], [[ 49.58872986, -21.63500786, 52.23811722], [ 31.22685432, -84.54801178, 26.00300598], [ 29.85455513, -40.19913864, 16.71471214], ..., [ -99.14375305, -107.97598267, -17.22589111], [-117.37517548, -144.86048889, -27.97831154], [ -24.96165657, -105.30418396, 20.67388725]]]], dtype=floa t32) Expected Output: **Iteration 0 : ** total cost = 5.05035e+09 content cost = 7877.67 style cost = 1.26257e+08 You're done! After running this, in the upper bar of the notebook click on "File" and then "Open". Go to the "/output" directory to see all the saved images. Open "generated_image" to see the generated image! :) You should see something the image presented below on the right: We didn't want you to wait too long to see an initial result, and so had set the hyperparameters accordingly. To get the best looking results, running the optimization algorithm longer (and perhaps with a smaller learning rate) might work better. After completing and submitting this assignment, we encourage you to come back and play more with this notebook, and see if you can generate even better looking images. Here are few other examples: The beautiful ruins of the ancient city of Persepolis (Iran) with the style of Van Gogh (The Starry Night) The tomb of Cyrus the great in Pasargadae with the style of a Ceramic Kashi from Ispahan. A scientific study of a turbulent fluid with the style of a abstract blue fluid painting. 5 - Test with your own image (Optional/Ungraded) Finally, you can also rerun the algorithm on your own images! To do so, go back to part 4 and change the content image and style image with your own pictures. In detail, here's what you should do: 1. Click on "File - Open" in the upper tab of the notebook Go to "/images" and upload your images (requirement: (WIDTH = 300, HEIGHT = 225)), rename them "my_content.png" and "my_style.png" for example. 2. Change the code in part (3.4) from : content_image = scipy.misc.imread("images/louvre.jpg") style_image = scipy.misc.imread("images/claude-monet.jpg") to: content_image = scipy.misc.imread("images/my_content.jpg") style_image = scipy.misc.imread("images/my_style.jpg") 3. 4. Rerun the cells (you may need to restart the Kernel in the upper tab of the notebook). You can also tune your hyperparameters: Which layers are responsible for representing the style? STYLE_LAYERS How many iterations do you want to run the algorithm? num_iterations What is the relative weighting between content and style? alpha/beta 6 - Conclusion Great job on completing this assignment! You are now able to use Neural Style Transfer to generate artistic images. This is also your first time building a model in which the optimization algorithm updates the pixel values rather than the neural network's parameters. Deep learning has many different types of models and this is only one of them! What you should remember: Neural Style Transfer is an algorithm that given a content image C and a style image S can generate an artistic image It uses representations (hidden layer activations) based on a pretrained ConvNet. The content cost function is computed using one hidden layer's activations. The style cost function for one layer is computed using the Gram matrix of that layer's activations. The overall style cost function is obtained using several hidden layers. Optimizing the total cost function results in synthesizing new images. This was the final programming exercise of this course. Congratulations--you've finished all the programming exercises of this course on Convolutional Networks! We hope to also see you in Course 5, on Sequence models! References: The Neural Style Transfer algorithm was due to Gatys et al. (2015). Harish Narayanan and Github user "log0" also have highly readable write-ups from which we drew inspiration. The pre-trained network used in this implementation is a VGG network, which is due to Simonyan and Zisserman (2015). Pre-trained weights were from the work of the MathConvNet team. Leon A. Gatys, Alexander S. Ecker, Matthias Bethge, (2015). A Neural Algorithm of Artistic Style (https://arxiv.org/abs/1508.06576 (https://arxiv.org/abs/1508.06576)) Harish Narayanan, Convolutional neural networks for artistic style transfer. https://harishnarayanan.org/writing/artistic-style-transfer/ (https://harishnarayanan.org/writing/artisticstyle-transfer/) Log0, TensorFlow Implementation of "A Neural Algorithm of Artistic Style". http://www.chioka.in /tensorflow-implementation-neural-algorithm-of-artistic-style (http://www.chioka.in/tensorflowimplementation-neural-algorithm-of-artistic-style) Karen Simonyan and Andrew Zisserman (2015). Very deep convolutional networks for large-scale image recognition (https://arxiv.org/pdf/1409.1556.pdf (https://arxiv.org/pdf/1409.1556.pdf)) MatConvNet. http://www.vlfeat.org/matconvnet/pretrained/ (http://www.vlfeat.org/matconvnet /pretrained/) In [ ]:
Welcome to the second assignment of this week. In this assignment, you will learn about Neural Style Transfer. This algorithm was created by Gatys et al. (2015) (https://arxiv.org/abs/1508.06576 (https://arxiv.org /abs/1508.06576)). In this assignment, you will: Implement the neural style transfer algorithm Generate novel artistic images using your algorithm Most of the algorithms you've studied optimize a cost function to get a set of parameter values. In Neural Style Transfer, you'll optimize a cost function to get pixel values! In [1]: import os import sys import scipy.io import scipy.misc import matplotlib.pyplot as plt from matplotlib.pyplot import imshow from PIL import Image from nst_utils import * import numpy as np import tensorflow as tf %matplotlib inline 1 - Problem Statement Neural Style Transfer (NST) is one of the most fun techniques in deep learning. As seen below, it merges two images, namely, a "content" image (C) and a "style" image (S), to create a "generated" image (G). The generated image G combines the "content" of the image C with the "style" of image S. In this example, you are going to generate an image of the Louvre museum in Paris (content image C), mixed with a painting by Claude Monet, a leader of the impressionist movement (style image S). Let's see how you can do this. 2 - Transfer Learning Neural Style Transfer (NST) uses a previously trained convolutional network, and builds on top of that. The idea of using a network trained on a different task and applying it to a new task is called transfer learning. Following the original NST paper (https://arxiv.org/abs/1508.06576 (https://arxiv.org/abs/1508.06576)), we will use the VGG network. Specifically, we'll use VGG-19, a 19-layer version of the VGG network. This model has already been trained on the very large ImageNet database, and thus has learned to recognize a variety of low level features (at the earlier layers) and high level features (at the deeper layers). Run the following code to load parameters from the VGG model. This may take a few seconds. In [2]: model = load_vgg_model("pretrained-model/imagenet-vgg-verydeep-19.mat") print(model) The model is stored in a python dictionary where each variable name is the key and the corresponding value is a tensor containing that variable's value. To run an image through this network, you just have to feed the image to the model. In TensorFlow, you can do so using the tf.assign (https://www.tensorflow.org/api_docs /python/tf/assign) function. In particular, you will use the assign function like this: model["input"].assign(image) This assigns the image as an input to the model. After this, if you want to access the activations of a particular layer, say layer 4_2 when the network is run on this image, you would run a TensorFlow session on the correct tensor conv4_2, as follows: sess.run(model["conv4_2"]) {'input': <tf.Variable 'Variable:0' shape=(1, 300, 400, 3) dtype=floa t32_ref, 'conv1_1': <tf.Tensor 'Relu:0' shape=(1, 300, 400, 64) dtyp e=float32, 'conv1_2': <tf.Tensor 'Relu_1:0' shape=(1, 300, 400, 64) dtype=float32, 'avgpool1': <tf.Tensor 'AvgPool:0' shape=(1, 150, 200 , 64) dtype=float32, 'conv2_1': <tf.Tensor 'Relu_2:0' shape=(1, 150, 200, 128) dtype=float32, 'conv2_2': <tf.Tensor 'Relu_3:0' shape=(1, 150, 200, 128) dtype=float32, 'avgpool2': <tf.Tensor 'AvgPool_1:0' s hape=(1, 75, 100, 128) dtype=float32, 'conv3_1': <tf.Tensor 'Relu_4: 0' shape=(1, 75, 100, 256) dtype=float32, 'conv3_2': <tf.Tensor 'Rel u_5:0' shape=(1, 75, 100, 256) dtype=float32, 'conv3_3': , 'conv3_4': , 'avgpool3': , 'conv4_ 1': <tf.Tensor 'Relu_8:0' shape=(1, 38, 50, 512) dtype=float32, 'con v4_2': <tf.Tensor 'Relu_9:0' shape=(1, 38, 50, 512) dtype=float32, ' conv4_3': <tf.Tensor 'Relu_10:0' shape=(1, 38, 50, 512) dtype=float32 , 'conv4_4': <tf.Tensor 'Relu_11:0' shape=(1, 38, 50, 512) dtype=flo at32, 'avgpool4': <tf.Tensor 'AvgPool_3:0' shape=(1, 19, 25, 512) dt ype=float32, 'conv5_1': <tf.Tensor 'Relu_12:0' shape=(1, 19, 25, 512 ) dtype=float32, 'conv5_2': <tf.Tensor 'Relu_13:0' shape=(1, 19, 25, 512) dtype=float32, 'conv5_3': <tf.Tensor 'Relu_14:0' shape=(1, 19, 25, 512) dtype=float32, 'conv5_4': <tf.Tensor 'Relu_15:0' shape=(1, 19, 25, 512) dtype=float32, 'avgpool5': <tf.Tensor 'AvgPool_4:0' sha pe=(1, 10, 13, 512) dtype=float32} 3 - Neural Style Transfer We will build the NST algorithm in three steps: Build the content cost function Build the style cost function Put it together to get . 3.1 - Computing the content cost In our running example, the content image C will be the picture of the Louvre Museum in Paris. Run the code below to see a picture of the Louvre. In [3]: content_image = scipy.misc.imread("images/louvre.jpg") imshow(content_image) Out[3]: The content image (C) shows the Louvre museum's pyramid surrounded by old Paris buildings, against a sunny sky with a few clouds. 3.1.1 - How do you ensure the generated image G matches the content of the image C? As we saw in lecture, the earlier (shallower) layers of a ConvNet tend to detect lower-level features such as edges and simple textures, and the later (deeper) layers tend to detect higher-level features such as more complex textures as well as object classes. We would like the "generated" image G to have similar content as the input image C. Suppose you have chosen some layer's activations to represent the content of an image. In practice, you'll get the most visually pleasing results if you choose a layer in the middle of the network--neither too shallow nor too deep. (After you have finished this exercise, feel free to come back and experiment with using different layers, to see how the results vary.) So, suppose you have picked one particular hidden layer to use. Now, set the image C as the input to the pretrained VGG network, and run forward propagation. Let be the hidden layer activations in the layer you had chosen. (In lecture, we had written this as , but here we'll drop the superscript to simplify the notation.) This will be a tensor. Repeat this process with the image G: Set G as the input, and run forward progation. Let be the corresponding hidden layer activation. We will define as the content cost function as: Here, and are the height, width and number of channels of the hidden layer you have chosen, and appear in a normalization term in the cost. For clarity, note that and are the volumes corresponding to a hidden layer's activations. In order to compute the cost , it might also be convenient to unroll these 3D volumes into a 2D matrix, as shown below. (Technically this unrolling step isn't needed to compute , but it will be good practice for when you do need to carry out a similar operation later for computing the style const .) In [4]: # GRADED FUNCTION: compute_content_cost def compute_content_cost(a_C, a_G): """ Computes the content cost Arguments: a_C -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing content of the image C a_G -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing content of the image G Returns: J_content -- scalar that you compute using equation 1 above. """ ### START CODE HERE ### # Retrieve dimensions from a_G (≈1 line) m, n_H, n_W, n_C = a_G.get_shape().as_list() # Reshape a_C and a_G (≈2 lines) a_C_unrolled = tf.reshape(a_C, [n_H * n_W, n_C]) a_G_unrolled = tf.reshape(a_G, [n_H * n_W, n_C]) # compute the cost with tensorflow (≈1 line) J_content = tf.reduce_sum(tf.square(tf.subtract(a_C_unrolled, a_G_ unrolled))) / (4 * n_H * n_W * n_C) ### END CODE HERE ### return J_content In [5]: tf.reset_default_graph() with tf.Session() as test: tf.set_random_seed(1) a_C = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) a_G = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) J_content = compute_content_cost(a_C, a_G) print("J_content = " + str(J_content.eval())) Expected Output: **J_content** 6.76559 What you should remember: The content cost takes a hidden layer activation of the neural network, and measures how different and are. When we minimize the content cost later, this will help make sure has similar content as . J_content = 6.76559 3.2 - Computing the style cost For our running example, we will use the following style image: In [6]: style_image = scipy.misc.imread("images/monet_800600.jpg") imshow(style_image) This painting was painted in the style of impressionism (https://en.wikipedia.org/wiki/Impressionism). Lets see how you can now define a "style" const function . Out[6]: 3.2.1 - Style matrix The style matrix is also called a "Gram matrix." In linear algebra, the Gram matrix G of a set of vectors is the matrix of dot products, whose entries are . In other words, compares how similar is to : If they are highly similar, you would expect them to have a large dot product, and thus for to be large. Note that there is an unfortunate collision in the variable names used here. We are following common terminology used in the literature, but is used to denote the Style matrix (or Gram matrix) as well as to denote the generated image . We will try to make sure which we are referring to is always clear from the context. In NST, you can compute the Style matrix by multiplying the "unrolled" filter matrix with their transpose: The result is a matrix of dimension where is the number of filters. The value measures how similar the activations of filter are to the activations of filter . One important part of the gram matrix is that the diagonal elements such as also measures how active filter is. For example, suppose filter is detecting vertical textures in the image. Then measures how common vertical textures are in the image as a whole: If is large, this means that the image has a lot of vertical texture. By capturing the prevalence of different types of features ( ), as well as how much different features occur together ( ), the Style matrix measures the style of an image. Exercise: Using TensorFlow, implement a function that computes the Gram matrix of a matrix A. The formula is: The gram matrix of A is . If you are stuck, take a look at Hint 1 (https://www.tensorflow.org /api_docs/python/tf/matmul) and Hint 2 (https://www.tensorflow.org/api_docs/python/tf/transpose). In [7]: # GRADED FUNCTION: gram_matrix def gram_matrix(A): """ Argument: A -- matrix of shape (n_C, n_H*n_W) Returns: GA -- Gram matrix of A, of shape (n_C, n_C) """ ### START CODE HERE ### (≈1 line) GA = tf.matmul(A, A, transpose_a=False, transpose_b=True,) ### END CODE HERE ### return GA In [8]: tf.reset_default_graph() with tf.Session() as test: tf.set_random_seed(1) A = tf.random_normal([3, 2*1], mean=1, stddev=4) GA = gram_matrix(A) print("GA = " + str(GA.eval())) Expected Output: **GA** [[ 6.42230511 -4.42912197 -2.09668207] [ -4.42912197 19.46583748 19.56387138] [ -2.09668207 19.56387138 20.6864624 ]] 3.2.2 - Style cost After generating the Style matrix (Gram matrix), your goal will be to minimize the distance between the Gram matrix of the "style" image S and that of the "generated" image G. For now, we are using only a single hidden layer , and the corresponding style cost for this layer is defined as: where and are respectively the Gram matrices of the "style" image and the "generated" image, computed using the hidden layer activations for a particular hidden layer in the network. GA = [[ 6.42230511 -4.42912197 -2.09668207] [ -4.42912197 19.46583748 19.56387138] [ -2.09668207 19.56387138 20.6864624 ]] Exercise: Compute the style cost for a single layer. Instructions: The 3 steps to implement this function are: Retrieve dimensions from the hidden layer activations a_G: To retrieve dimensions from a tensor X, use: X.get_shape().as_list() 1. Unroll the hidden layer activations a_S and a_G into 2D matrices, as explained in the picture above. You may find Hint1 (https://www.tensorflow.org/versions/r1.3/api_docs/python/tf/transpose) and Hint2 (https://www.tensorflow.org/versions/r1.2/api_docs/python/tf/reshape) useful. 2. 3. Compute the Style matrix of the images S and G. (Use the function you had previously written.) Compute the Style cost: You may find Hint3 (https://www.tensorflow.org/api_docs/python/tf/reduce_sum), Hint4 (https://www.tensorflow.org/api_docs/python/tf/square) and Hint5 (https://www.tensorflow.org/api_docs/python/tf/subtract) useful. 4. In [9]: # GRADED FUNCTION: compute_layer_style_cost def compute_layer_style_cost(a_S, a_G): """ Arguments: a_S -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing style of the image S a_G -- tensor of dimension (1, n_H, n_W, n_C), hidden layer activat ions representing style of the image G Returns: J_style_layer -- tensor representing a scalar value, style cost def ined above by equation (2) """ ### START CODE HERE ### # Retrieve dimensions from a_G (≈1 line) m, n_H, n_W, n_C = a_G.get_shape().as_list() # Reshape the images to have them of shape (n_C, n_H*n_W) (≈2 lines ) a_S = tf.transpose(tf.reshape(a_S, [n_H * n_W, n_C])) a_G = tf.transpose(tf.reshape(a_G, [n_H * n_W, n_C])) # Computing gram_matrices for both images S and G (≈2 lines) GS = gram_matrix(a_S) GG = gram_matrix(a_G) # Computing the loss (≈1 line) J_style_layer = tf.reduce_sum(tf.square(tf.subtract(GS, GG))) / (4 * np.square(n_C) * np.square(n_W * n_H) ) ### END CODE HERE ### return J_style_layer In [10]: tf.reset_default_graph() with tf.Session() as test: tf.set_random_seed(1) a_S = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) a_G = tf.random_normal([1, 4, 4, 3], mean=1, stddev=4) J_style_layer = compute_layer_style_cost(a_S, a_G) print("J_style_layer = " + str(J_style_layer.eval())) Expected Output: **J_style_layer** 9.19028 3.2.3 Style Weights So far you have captured the style from only one layer. We'll get better results if we "merge" style costs from several different layers. After completing this exercise, feel free to come back and experiment with different weights to see how it changes the generated image . But for now, this is a pretty reasonable default: In [11]: STYLE_LAYERS = [ ('conv1_1', 0.2), ('conv2_1', 0.2), ('conv3_1', 0.2), ('conv4_1', 0.2), ('conv5_1', 0.2)] You can combine the style costs for different layers as follows: where the values for are given in STYLE_LAYERS. We've implemented a compute_style_cost(...) function. It simply calls your compute_layer_style_cost(...) several times, and weights their results using the values in STYLE_LAYERS. Read over it to make sure you understand what it's doing. J_style_layer = 9.19028 In [12]: def compute_style_cost(model, STYLE_LAYERS): """ Computes the overall style cost from several chosen layers Arguments: model -- our tensorflow model STYLE_LAYERS -- A python list containing: - the names of the layers we would like to extr act style from - a coefficient for each of them Returns: J_style -- tensor representing a scalar value, style cost defined a bove by equation (2) """ # initialize the overall style cost J_style = 0 for layer_name, coeff in STYLE_LAYERS: # Select the output tensor of the currently selected layer out = model[layer_name] # Set a_S to be the hidden layer activation from the layer we h ave selected, by running the session on out a_S = sess.run(out) # Set a_G to be the hidden layer activation from same layer. He re, a_G references model[layer_name] # and isn't evaluated yet. Later in the code, we'll assign the image G as the model input, so that # when we run the session, this will be the activations drawn f rom the appropriate layer, with G as input. a_G = out # Compute style_cost for the current layer J_style_layer = compute_layer_style_cost(a_S, a_G) # Add coeff * J_style_layer of this layer to overall style cost J_style += coeff * J_style_layer return J_style Note: In the inner-loop of the for-loop above, a_G is a tensor and hasn't been evaluated yet. It will be evaluated and updated at each iteration when we run the TensorFlow graph in model_nn() below. What you should remember: The style of an image can be represented using the Gram matrix of a hidden layer's activations. However, we get even better results combining this representation from multiple different layers. This is in contrast to the content representation, where usually using just a single hidden layer is sufficient. Minimizing the style cost will cause the image to follow the style of the image . </font color='blue' 3.3 - Defining the total cost to optimize Finally, let's create a cost function that minimizes both the style and the content cost. The formula is: Exercise: Implement the total cost function which includes both the content cost and the style cost. In [13]: # GRADED FUNCTION: total_cost def total_cost(J_content, J_style, alpha = 10, beta = 40): """ Computes the total cost function Arguments: J_content -- content cost coded above J_style -- style cost coded above alpha -- hyperparameter weighting the importance of the content cos t beta -- hyperparameter weighting the importance of the style cost Returns: J -- total cost as defined by the formula above. """ ### START CODE HERE ### (≈1 line) J = (alpha * J_content) + (beta * J_style) ### END CODE HERE ### return J In [14]: tf.reset_default_graph() with tf.Session() as test: np.random.seed(3) J_content = np.random.randn() J_style = np.random.randn() J = total_cost(J_content, J_style) print("J = " + str(J)) Expected Output: **J** 35.34667875478276 What you should remember: The total cost is a linear combination of the content cost and the style cost and are hyperparameters that control the relative weighting between content and style 4 - Solving the optimization problem Finally, let's put everything together to implement Neural Style Transfer! Here's what the program will have to do: 1. Create an Interactive Session 2. Load the content image 3. Load the style image 4. Randomly initialize the image to be generated 5. Load the VGG16 model Build the TensorFlow graph: Run the content image through the VGG16 model and compute the content cost Run the style image through the VGG16 model and compute the style cost Compute the total cost Define the optimizer and the learning rate 6. Initialize the TensorFlow graph and run it for a large number of iterations, updating the generated image at every step. 7. Lets go through the individual steps in detail. J = 35.34667875478276 You've previously implemented the overall cost . We'll now set up TensorFlow to optimize this with respect to . To do so, your program has to reset the graph and use an "Interactive Session (https://www.tensorflow.org/api_docs/python/tf/InteractiveSession)". Unlike a regular session, the "Interactive Session" installs itself as the default session to build a graph. This allows you to run variables without constantly needing to refer to the session object, which simplifies the code. Lets start the interactive session. In [15]: # Reset the graph tf.reset_default_graph() # Start interactive session sess = tf.InteractiveSession() Let's load, reshape, and normalize our "content" image (the Louvre museum picture): In [16]: content_image = scipy.misc.imread("images/louvre_small.jpg") content_image = reshape_and_normalize_image(content_image) Let's load, reshape and normalize our "style" image (Claude Monet's painting): In [17]: style_image = scipy.misc.imread("images/monet.jpg") style_image = reshape_and_normalize_image(style_image) Now, we initialize the "generated" image as a noisy image created from the content_image. By initializing the pixels of the generated image to be mostly noise but still slightly correlated with the content image, this will help the content of the "generated" image more rapidly match the content of the "content" image. (Feel free to look in nst_utils.py to see the details of generate_noise_image(...); to do so, click "File--Open..." at the upper-left corner of this Jupyter notebook.) In [18]: generated_image = generate_noise_image(content_image) imshow(generated_image[0]) Next, as explained in part (2), let's load the VGG16 model. In [19]: model = load_vgg_model("pretrained-model/imagenet-vgg-verydeep-19.mat") To get the program to compute the content cost, we will now assign a_C and a_G to be the appropriate hidden layer activations. We will use layer conv4_2 to compute the content cost. The code below does the following: 1. Assign the content image to be the input to the VGG model. 2. Set a_C to be the tensor giving the hidden layer activation for layer "conv4_2". 3. Set a_G to be the tensor giving the hidden layer activation for the same layer. 4. Compute the content cost using a_C and a_G. Out[18]: In [20]: # Assign the content image to be the input of the VGG model. sess.run(model['input'].assign(content_image)) # Select the output tensor of layer conv4_2 out = model['conv4_2'] # Set a_C to be the hidden layer activation from the layer we have sele cted a_C = sess.run(out) # Set a_G to be the hidden layer activation from same layer. Here, a_G references model['conv4_2'] # and isn't evaluated yet. Later in the code, we'll assign the image G as the model input, so that # when we run the session, this will be the activations drawn from the appropriate layer, with G as input. a_G = out # Compute the content cost J_content = compute_content_cost(a_C, a_G) Note: At this point, a_G is a tensor and hasn't been evaluated. It will be evaluated and updated at each iteration when we run the Tensorflow graph in model_nn() below. In [21]: # Assign the input of the model to be the "style" image sess.run(model['input'].assign(style_image)) # Compute the style cost J_style = compute_style_cost(model, STYLE_LAYERS) Exercise: Now that you have J_content and J_style, compute the total cost J by calling total_cost(). Use alpha = 10 and beta = 40. In [22]: ### START CODE HERE ### (1 line) J = total_cost(J_content, J_style, alpha = 10, beta = 40) ### END CODE HERE ### You'd previously learned how to set up the Adam optimizer in TensorFlow. Lets do that here, using a learning rate of 2.0. See reference (https://www.tensorflow.org/api_docs/python/tf/train/AdamOptimizer) In [23]: # define optimizer (1 line) optimizer = tf.train.AdamOptimizer(2.0) # define train_step (1 line) train_step = optimizer.minimize(J) Exercise: Implement the model_nn() function which initializes the variables of the tensorflow graph, assigns the input image (initial generated image) as the input of the VGG16 model and runs the train_step for a large number of steps. In [32]: def model_nn(sess, input_image, num_iterations = 200): # Initialize global variables (you need to run the session on the i nitializer) ### START CODE HERE ### (1 line) sess.run(tf.global_variables_initializer()) ### END CODE HERE ### # Run the noisy input image (initial generated image) through the m odel. Use assign(). ### START CODE HERE ### (1 line) sess.run(model['input'].assign(input_image)) ### END CODE HERE ### for i in range(num_iterations): # Run the session on the train_step to minimize the total cost ### START CODE HERE ### (1 line) sess.run(train_step) ### END CODE HERE ### # Compute the generated image by running the session on the cur rent model['input'] ### START CODE HERE ### (1 line) generated_image = sess.run(model['input']) ### END CODE HERE ### # Print every 20 iteration. if i%20 == 0: Jt, Jc, Js = sess.run([J, J_content, J_style]) print("Iteration " + str(i) + " :") print("total cost = " + str(Jt)) print("content cost = " + str(Jc)) print("style cost = " + str(Js)) # save current generated image in the "/output" directory save_image("output/" + str(i) + ".png", generated_image) # save last generated image save_image('output/generated_image.jpg', generated_image) return generated_image Run the following cell to generate an artistic image. It should take about 3min on CPU for every 20 iterations but you start observing attractive results after ≈140 iterations. Neural Style Transfer is generally trained using GPUs. In [33]: model_nn(sess, generated_image) Iteration 0 : total cost = 5.05035e+09 content cost = 7877.67 style cost = 1.26257e+08 Iteration 20 : total cost = 9.43282e+08 content cost = 15187.1 style cost = 2.35783e+07 Iteration 40 : total cost = 4.84914e+08 content cost = 16785.8 style cost = 1.21187e+07 Iteration 60 : total cost = 3.12578e+08 content cost = 17465.4 style cost = 7.81007e+06 Iteration 80 : total cost = 2.28117e+08 content cost = 17715.0 style cost = 5.69849e+06 Iteration 100 : total cost = 1.80678e+08 content cost = 17897.0 style cost = 4.51247e+06 Iteration 120 : total cost = 1.49979e+08 content cost = 18036.8 style cost = 3.74497e+06 Iteration 140 : total cost = 1.27725e+08 content cost = 18192.9 style cost = 3.18859e+06 Iteration 160 : total cost = 1.107e+08 content cost = 18353.7 style cost = 2.76292e+06 Iteration 180 : total cost = 9.73286e+07 content cost = 18498.0 style cost = 2.42859e+06 Out[33]: array([[[[ -47.79544449, -61.47901154, 48.77702713], [ -26.42229462, -40.57945633, 26.92292023], [ -41.86124039, -28.98380089, 11.30953312], ..., [ -26.89013481, -9.54228497, 14.28584766], [ -30.29121208, -2.94760489, 24.04436874], [ -42.41747665, -4.17948103, 49.22496033]], [[ -61.32366562, -51.78530502, 25.22885132], [ -33.15329361, -31.11307716, -1.55355597], [ -27.22502708, -30.70824814, 15.15648651], ..., [ -26.75429535, -4.98880053, 26.1478672 ], [ -21.36663246, -16.82632828, 14.10803032], [ -40.67829895, -6.07156563, 9.60023594]], [[ -52.3535614 , -51.64604187, 13.5016737 ], [ -37.27353668, -41.49959183, -6.35776997], [ -34.27314377, -25.33574867, 7.28916264], ..., [ -10.58620262, -37.16160965, 12.67485428], [ -12.21832657, -20.8809433 , 17.26901627], [ -22.42559814, -18.40380859, 14.41193676]], ..., [[ -49.01599503, -53.96035004, -37.23251724], [ -97.29430389, -76.19354248, -267.10494995], [ -74.4513092 , -70.41262054, -141.00666809], ..., [ -69.40086365, -70.02838898, -29.45741272], [ -78.91752625, -87.95886993, -22.85899162], [ 1.8733685 , -39.92020798, 23.59747505]], [[ -1.7835474 , -74.61296082, 13.84414291], [-172.76348877, -101.91304016, -30.54630089], [ 7.31526756, -70.08149719, -20.32836342], ..., [ -95.6863327 , -84.15804291, -47.7417717 ], [-102.43443298, -102.8863678 , -59.78057098], [ -65.53679657, -95.97616577, 1.40922606]], [[ 49.58872986, -21.63500786, 52.23811722], [ 31.22685432, -84.54801178, 26.00300598], [ 29.85455513, -40.19913864, 16.71471214], ..., [ -99.14375305, -107.97598267, -17.22589111], [-117.37517548, -144.86048889, -27.97831154], [ -24.96165657, -105.30418396, 20.67388725]]]], dtype=floa t32) Expected Output: **Iteration 0 : ** total cost = 5.05035e+09 content cost = 7877.67 style cost = 1.26257e+08 You're done! After running this, in the upper bar of the notebook click on "File" and then "Open". Go to the "/output" directory to see all the saved images. Open "generated_image" to see the generated image! :) You should see something the image presented below on the right: We didn't want you to wait too long to see an initial result, and so had set the hyperparameters accordingly. To get the best looking results, running the optimization algorithm longer (and perhaps with a smaller learning rate) might work better. After completing and submitting this assignment, we encourage you to come back and play more with this notebook, and see if you can generate even better looking images. Here are few other examples: The beautiful ruins of the ancient city of Persepolis (Iran) with the style of Van Gogh (The Starry Night) The tomb of Cyrus the great in Pasargadae with the style of a Ceramic Kashi from Ispahan. A scientific study of a turbulent fluid with the style of a abstract blue fluid painting. 5 - Test with your own image (Optional/Ungraded) Finally, you can also rerun the algorithm on your own images! To do so, go back to part 4 and change the content image and style image with your own pictures. In detail, here's what you should do: 1. Click on "File - Open" in the upper tab of the notebook Go to "/images" and upload your images (requirement: (WIDTH = 300, HEIGHT = 225)), rename them "my_content.png" and "my_style.png" for example. 2. Change the code in part (3.4) from : content_image = scipy.misc.imread("images/louvre.jpg") style_image = scipy.misc.imread("images/claude-monet.jpg") to: content_image = scipy.misc.imread("images/my_content.jpg") style_image = scipy.misc.imread("images/my_style.jpg") 3. 4. Rerun the cells (you may need to restart the Kernel in the upper tab of the notebook). You can also tune your hyperparameters: Which layers are responsible for representing the style? STYLE_LAYERS How many iterations do you want to run the algorithm? num_iterations What is the relative weighting between content and style? alpha/beta 6 - Conclusion Great job on completing this assignment! You are now able to use Neural Style Transfer to generate artistic images. This is also your first time building a model in which the optimization algorithm updates the pixel values rather than the neural network's parameters. Deep learning has many different types of models and this is only one of them! What you should remember: Neural Style Transfer is an algorithm that given a content image C and a style image S can generate an artistic image It uses representations (hidden layer activations) based on a pretrained ConvNet. The content cost function is computed using one hidden layer's activations. The style cost function for one layer is computed using the Gram matrix of that layer's activations. The overall style cost function is obtained using several hidden layers. Optimizing the total cost function results in synthesizing new images. This was the final programming exercise of this course. Congratulations--you've finished all the programming exercises of this course on Convolutional Networks! We hope to also see you in Course 5, on Sequence models! References: The Neural Style Transfer algorithm was due to Gatys et al. (2015). Harish Narayanan and Github user "log0" also have highly readable write-ups from which we drew inspiration. The pre-trained network used in this implementation is a VGG network, which is due to Simonyan and Zisserman (2015). Pre-trained weights were from the work of the MathConvNet team. Leon A. Gatys, Alexander S. Ecker, Matthias Bethge, (2015). A Neural Algorithm of Artistic Style (https://arxiv.org/abs/1508.06576 (https://arxiv.org/abs/1508.06576)) Harish Narayanan, Convolutional neural networks for artistic style transfer. https://harishnarayanan.org/writing/artistic-style-transfer/ (https://harishnarayanan.org/writing/artisticstyle-transfer/) Log0, TensorFlow Implementation of "A Neural Algorithm of Artistic Style". http://www.chioka.in /tensorflow-implementation-neural-algorithm-of-artistic-style (http://www.chioka.in/tensorflowimplementation-neural-algorithm-of-artistic-style) Karen Simonyan and Andrew Zisserman (2015). Very deep convolutional networks for large-scale image recognition (https://arxiv.org/pdf/1409.1556.pdf (https://arxiv.org/pdf/1409.1556.pdf)) MatConvNet. http://www.vlfeat.org/matconvnet/pretrained/ (http://www.vlfeat.org/matconvnet /pretrained/) In [ ]:
1 file (14.2MB)
