search

rrmina / fast-neural-style-pytorch

Fast Neural Style Transfer implementation in PyTorch :art: :art: :art:
315 stars 77 forks source link

Custom model #16

Open tolchx opened 4 years ago

tolchx commented 4 years ago

Hello, very good work, I was able to make it work without problem. I want to try creating a model, using a different style, some other painting or artist. How can I create my own model / vgg16-00b39a1b.pth?

If you have any guide to experiment with creating models it would be of great help, I will continue investigating, thanks

rrmina commented 4 years ago

Hi. Thanks!

Have a look at this notebook: https://nbviewer.jupyter.org/github/rrmina/fast-neural-style-pytorch/blob/master/notebook/mosaic_TransformerNetwork.ipynb

All you need to do is to change the style image.