MrinalJain17 / neural-style-transfer

Implementation of Neural Style Transfer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Neural Style Transfer

The following papers are implemented in this repository:

  1. A Neural Algorithm of Artistic Style
  2. Improving the Neural Algorithm of Artistic Style
  3. Preserving Color in Neural Artistic Style Transfer
  4. Perceptual Losses for Real-Time Style Transfer and Super-Resolution

Examples

example_1


example_2


example_3


We have setup an interactive report with a bunch of other visualizations - Link.

Installation

Execute the following command from the root of the repository to install the project:

    pip install -e .

Note that this step is required to run the project.

Requirements

Base Requirements

  1. Python (3.7)
  2. PyTorch (1.7)
  3. Torchvision (0.8)
  4. Tqdm - For displaying progress bars
  5. Weights and Biases - For visualization

Additional Requirements

PyTorch-Lightning (1.0) is required for executing Fast Style Transfer.

References

  1. Neural Style Transfer by Somshubra Majumdar

About

Implementation of Neural Style Transfer

License:MIT License


Languages

Language:Python 100.0%