PyTorch-CycleGAN: A Clean and Readable PyTorch Implementation of CycleGAN

PyTorch-CycleGAN: A Clean and Readable PyTorch Implementation of CycleGAN

License: GPL-3.0
Model Type: Generative AI
This repository offers a straightforward and well-documented implementation of CycleGAN in PyTorch. CycleGAN is a deep learning model that enables image-to-image translation without paired data, making it ideal for tasks like photo enhancement, style transfer, and domain adaptation.

Key Features

  • Unpaired Image Translation: Facilitates transformation between two image domains without the need for paired datasets.
  • Modular Codebase: Organized structure with separate files for models, datasets, utilities, and training scripts.
  • Visdom Integration: Supports real-time visualization of training progress and generated images via Visdom.
  • Predefined Datasets: Includes scripts to download and set up popular datasets such as horse2zebra, monet2photo, and summer2winter_yosemite.
  • Training and Testing Scripts: Provides clear commands for training and testing the model, along with options for GPU acceleration.

Project Screenshots

Project Screenshot
Project Screenshot