Изменения
→Пример кода на Python
== Пример кода на Python ==
'''Content LossФункция потери контента'''
'''class''' ContentLoss(nn.Module):
return input
'''Style LossФункция потери стиля'''
'''def''' gram_matrix(input):
a, b, c, d = input.size() # a=batch size(=1)
return input
'''Importing the ModelИнициализация модели'''
cnn = models.vgg19(pretrained=True).features.to(device).eval()
'''NormalizationНормализация'''
cnn_normalization_mean = torch.tensor([0.485, 0.456, 0.406]).to(device)
cnn_normalization_std = torch.tensor([0.229, 0.224, 0.225]).to(device)
return (img - self.mean) / self.std
'''Adding our content loss and style loss layersДобавление собственных слоев'''
<font color="green"># desired depth layers to compute style/content losses :</font>
content_layers_default = ['conv_4']
return model, style_losses, content_losses
'''Gradient DescentГрадиентный спуск'''
'''def''' get_input_optimizer(input_img):
<font color="green"># this line to show that input is a parameter that requires a gradient</font>
return optimizer
'''Run algorithmЗапуск алгоритма'''
'''def''' run_style_transfer(cnn, normalization_mean, normalization_std,
content_img, style_img, input_img, num_steps=300,
