#133 · Primary category: Image Generation

PyTorch-CycleGAN

artificial-intelligence computer-graphics computer-vision cyclegan deep-learning generative-adversarial-network image-generation image-processing pytorch

A clean and readable Pytorch implementation of CycleGAN

Project last updated:04/20/22

GitHub Stars

1.3K

Forks

296

Contributors

1

License

GPL-3.0

Why we included this project

CycleGAN translates images between two domains without paired examples, turning horses into zebras or photos into paintings, and this repository is a straightforward way to study how that works. The code favors readability over raw speed, so you can trace the two generators and two discriminators and watch how the cycle-consistency loss keeps the round trips consistent. Setup stays light: a script pulls down the standard UC Berkeley datasets, including horse to zebra, apple to orange, and photo to Monet, and the training command starts with the hyperparameters the CycleGAN authors found to work well. You can also point it at your own folder structure if you have a custom style-transfer or domain-adaptation idea in mind. Just know the code targets Python 3.6 and the PyTorch of that era, so treat it as a learning and experimentation base rather than a production pipeline.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category