数据少也能训出高质量 GAN，可微增强一招搞定
data-efficient-gans 是 NeurIPS 2020 论文《Differentiable Augmentation for Data-Efficient GAN Training》的官方实现，旨在解决 GAN 训练数据不足时容易过拟合、训练不稳定、生成质量差的问题。其核心创新是提出可微增强（Differentiable Augmentation）方法，在判别器输入端应用多种图像变换（如平移、裁剪、颜色抖动等），并确保增强操作可微，使梯度能正常回传，从而在不增加真实数据的情况下显著提升生成器的泛化能力。项目提供 PyTorch 和 TensorFlow 两种框架的实现，支持多种数据集（如 CIFAR-10、FFHQ、Animal Faces 等）和多种 GAN 架构（如 StyleGAN2、BigGAN、SNGAN），并附带预训练模型和训练配置。通过简单修改现有 GAN 代码即可集成该方法，在少量数据（如 1000 张图）下也能训练出高质量的生成模型，是数据稀缺场景下训练 GAN 的实用工具。
⚠ 该项目 README 未提供完整安装说明，以下步骤为通用流程，请以官方文档为准。
1克隆仓库
把官方仓库拉到本地，后续所有 PyTorch/TensorFlow 实现都在这份代码的子目录里。
git clone https://github.com/mit-han-lab/data-efficient-gans.git2进入项目目录
切换到克隆下来的仓库根目录，查看 DiffAugment-stylegan2 等子目录结构。
cd data-efficient-gans3选择框架版本
PyTorch 版在 DiffAugment-stylegan2-pytorch 目录，TensorFlow 版在 DiffAugment-stylegan2 目录，二者按需选择。
4安装依赖
README 节选未给出具体安装命令，请打开所选子目录内的 README，按其说明安装依赖。
5获取数据集
README 提供了打包好的数据集下载链接，约 100 张图的训练设置可直接复现。
6用 Colab 试跑
官方 Colab 教程已发布，可在浏览器中直接体验 DiffAugment，无需本地配置环境。
README 节选未给出启动命令，可先跑通官方 Colab 教程或所选子目录 README 中的示例，确认环境可用。
Q：本地没有 GPU 能跑吗？
A：README 未说明硬件要求，但提供了官方 Colab 教程，可先在 Colab 上体验，再决定是否装本地环境。
Q：PyTorch 和 TensorFlow 版本选哪个？
A：PyTorch 实现在 DiffAugment-stylegan2-pytorch 目录，TensorFlow 实现在 DiffAugment-stylegan2 目录，按自己熟悉的框架选一个即可。
Q：训练数据从哪里来？
A：README 顶部提供了 datasets 的 Dropbox 打包下载链接，可直接下载使用。
Q：想训练 FFHQ 怎么办？
A：README 标注 FFHQ 训练已支持，具体做法见 DiffAugment-stylegan2 子目录的 README 中 FFHQ 一节。
StyleGAN2-ADA、FastGAN、GAN-Training-with-Data-Augmentation
data-efficient-gans 是图像领域的开源项目，由 mit-han-lab 开发，2020 年首次发布。
在全站 13,014 个收录项目中，它的 GitHub 星标数（1,308）位列前 30%，在图像分类的 837 个项目里位列前 13%。
项目已超过三个月没有代码更新，维护节奏明显放缓，最近一次代码更新于 2026-09-22。BSD-2-Clause许可证，完全免费，可自行部署使用。
它主要面向的使用场景是：小样本图像生成（如百张图训练GAN）。同类可对比的替代方案包括 StyleGAN2-ADA、FastGAN、GAN-Training-with-Data-Augmentation。
上一篇：TheaterGen
下一篇：PhotoPoster
satori
开源
网页代码一键变高清图，放大不糊，不用截图
Enlightened library to convert HTML and CSS to SVG
C2GAN
开源
用关键点精准控制图像生成，让AI画出你想要的结构
[ACM MM 2019 Oral] Cycle In Cycle Generative Adversarial Networks for Keypoint-Guide···
stable-diffusion-webui
开源
打开网页说想法，配图素材几秒到手
Stable Diffusion web UI
ComfyUI
开源
像搭积木一样连节点，生图流程自己搭
The most powerful and modular diffusion model GUI, api and backend with a graph/node···