Skip to content
Neural style transfer as a class in PyTorch
Python
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
00_styletransfernet.py removed broken optimizer Apr 3, 2017
README.md Update README.md Apr 3, 2017
acropolis_square_large.jpg
micromegas_square.jpg initial commit Apr 3, 2017
vgg.py

README.md

pt-styletransfer

Neural style transfer as a class in PyTorch

Based on: https://github.com/alexis-jacq/Pytorch-Tutorials

Adds:

StyleTransferNet as a class that can be imported by other scripts

VGG support (this was before pretrained VGG models were available in PyTorch)

Convenience functions for saving intermediate style & content targets for display

Convenience function for examining gram matricies as images

Auto style, content, and product image saving

matplotlib plots of losses over time & record of hyperparameters to keep track of favorable results

You can’t perform that action at this time.