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

---

---

---

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