Skip to content
This repository has been archived by the owner on Jan 3, 2020. It is now read-only.

Latest commit

 

History

History
21 lines (14 loc) · 1.57 KB

README.md

File metadata and controls

21 lines (14 loc) · 1.57 KB

Neural Style

style transfer with mxnet

Requirements

  • Install skimage via pip: sudo pip install scikit-image
  • Install CUDA and Cudnn
  • Install Mxnet, please set "USE_NVRTC = 1" in config.mk before compiling
  • Download pretrained VGG model and save it to the root of this repository.
  • Download MSCOCO dataset if you want to train models.

Usage

Folder mrf_cnn implements Combining Markov Random Fields and Convolutional Neural Networks for Image Synthesis. It is an optimization based method, can provide delicate result but is slow. Original Torch version

Folder perceptual implements Perceptual Losses for Real-Time Style Transfer and Super-Resolution. It trains a network to do optimization and is very fast. Original Torch version

Folder texturenet implements Texture Networks: Feed-forward Synthesis of Textures and Stylized Images. It also trains a network to do optimization. Original Torch version

Folder fast_mrf_cnn trains network to do optimization in mrf_cnn. It is very fast and can give result similar to but not as good as mrf_cnn.

Folder old_stuff contains some pretrained texture network models.