mimicry  by kwotsin

PyTorch library for reproducible GAN research

created 5 years ago
608 stars

Top 54.7% on sourcepulse

GitHubView on GitHub
Project Summary

Mimicry is a PyTorch library designed to address the reproducibility crisis in Generative Adversarial Network (GAN) research. It provides standardized implementations of popular GAN architectures, baseline scores trained and evaluated under consistent conditions, and a framework for researchers to focus on novel GAN implementations without boilerplate code. The library is beneficial for researchers and practitioners seeking to compare GANs fairly and ensure the reliability of reported results.

How It Works

Mimicry offers a unified framework for implementing, training, and evaluating various GAN models. It standardizes training procedures, hyperparameter choices, and evaluation metrics (FID, IS, KID) to facilitate direct comparisons between different GAN architectures. The library's core advantage lies in its curated model zoo and baseline results, which are verified against literature to ensure reproducibility. This approach allows users to quickly benchmark new GAN ideas against established performance levels.

Quick Start & Requirements

  • Install via pip: pip install git+https://github.com/kwotsin/mimicry.git
  • Requires PyTorch and a CUDA-enabled GPU for efficient training.
  • Official documentation and tutorials are available for detailed guidance.

Highlighted Details

  • Reproduces reported scores for popular GANs like SNGAN, WGAN-GP, and cGAN-PD.
  • Provides baseline results across multiple datasets (CIFAR-10, ImageNet, CelebA, LSUN-Bedroom, STL-10) with standard metrics.
  • Includes a model zoo with checkpoints for various GAN architectures and resolutions.
  • Supports custom GAN implementation and evaluation within its framework.

Maintenance & Community

The project is associated with the CVPR 2020 Workshop on AI for Content Creation. Further details on community engagement or active maintenance are not explicitly stated in the README.

Licensing & Compatibility

The project's license is not explicitly stated in the README. Compatibility for commercial use or closed-source linking would require clarification of the licensing terms.

Limitations & Caveats

The README does not specify any explicit limitations or known issues. However, as with many research-oriented libraries, the focus is on reproducing specific results, and broader applicability or robustness across all potential use cases may require further investigation.

Health Check
Last commit

3 years ago

Responsiveness

Inactive

Pull Requests (30d)
0
Issues (30d)
0
Star History
1 stars in the last 90 days

Explore Similar Projects

Starred by Aravind Srinivas Aravind Srinivas(Cofounder of Perplexity), Lilian Weng Lilian Weng(Cofounder of Thinking Machines Lab), and
9 more.

ganhacks by soumith

0.1%
12k
GAN training tips and tricks
created 8 years ago
updated 3 years ago
Starred by Ross Taylor Ross Taylor(Cofounder of General Reasoning; Creator of Papers with Code), Thomas Wolf Thomas Wolf(Cofounder of Hugging Face), and
1 more.

AdversarialNetsPapers by zhangqianhui

0.1%
7k
Paper list for generative adversarial networks (GANs)
created 8 years ago
updated 2 years ago
Feedback? Help us improve.