05_02_neural_style_transfer_train_and_analysis

g-i-o-r-g-i-o · Dec 15, 2019 · 49bd308 · 49bd308
1 parent bd01497
commit 49bd308
Show file tree

Hide file tree

Showing 5 changed files with 406 additions and 0 deletions.
diff --git a/05_02_neural_style_transfer_train_and_analysis.ipynb b/05_02_neural_style_transfer_train_and_analysis.ipynb
diff --git a/README.md b/README.md
@@ -25,6 +25,7 @@ Part 1: Introduction to Generative Deep Learning
 - 04_02_wgan_camel_analysis.ipynb
 - 04_02_wgan_cifar_analysis.ipynb
 - 04_03_wgangp_cifar_train_and_analysis.ipynb
+- 05_02_neural_style_transfer_and_analysis.ipynb
 
 ## requirements.txt
 

diff --git a/data_small/neural_style_transfer/content.jpg b/data_small/neural_style_transfer/content.jpg
diff --git a/data_small/neural_style_transfer/style.jpg b/data_small/neural_style_transfer/style.jpg
diff --git a/models/torch_commons.py b/models/torch_commons.py
@@ -2,6 +2,9 @@
 # from fastai https://github.com/fastai/fastai/blob/master/fastai/torch_core.py#L87
 import torch.nn as nn
 import functools
+import torch as t
+import torch.nn.functional as F
+from torchvision import transforms, models
 
 class PrePostInitMeta(type):
     "A metaclass that calls optional `__pre_init__` and `__post_init__` methods"