Calculating the losses

We now need the losses between the contents and styles of the two images. We will be using the mean squared loss as follows. Notice here that the subtraction in image1 - image2 is element-wise between the two image arrays. This subtraction works because the images have been resized to the same size in load_image:

def rms_loss(image1,image2):    loss = tf.reduce_mean(input_tensor=tf.square(image1 - image2))    return loss

So next, we define our content_loss function. This is just the mean squared difference between what is named content and target in the function signature:

def content_loss(content, target):  return rms_loss(content, target)

The style loss is defined in terms of a quantity called a Gram matrix. A Gram matrix, also ...

Get TensorFlow 2.0 Quick Start Guide now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.