https://github.com/tymokvo/pt-styletransfer

Neural style transfer as a class in PyTorch
https://github.com/tymokvo/pt-styletransfer

Last synced: 29 days ago
JSON representation

Neural style transfer as a class in PyTorch

• Host: GitHub
• URL: https://github.com/tymokvo/pt-styletransfer
• Owner: tymokvo
• Created: 2017-04-03T13:48:09.000Z (about 8 years ago)
• Default Branch: master
• Last Pushed: 2017-04-03T14:13:32.000Z (about 8 years ago)
• Last Synced: 2024-11-14T03:34:27.927Z (7 months ago)
• Language: Python
• Size: 644 KB
• Stars: 35
• Watchers: 3
• Forks: 4
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• Awesome-pytorch-list-CNVersion - pt-styletransfer
• Awesome-pytorch-list - pt-styletransfer

README

        
# 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