• This repository has been archived on 03/Jan/2020
  • Stars
    star
    119
  • Rank 296,204 (Top 6 %)
  • Language
    Python
  • Created about 8 years ago
  • Updated over 7 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

neural style transfer with mxnet

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.