zhaw / neural_style

neural style transfer with mxnet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

neural style transfer with mxnet


Languages

Language:Python 100.0%