unetgan

用 U-Net 判别器提升 GAN 生成细节,训练更稳、图像更真。

UNetGAN 是 CVPR 2020 论文《A U-Net Based Discriminator for Generative Adversarial Networks》的官方实现。它针对传统 GAN 判别器仅输出单一标量、难以捕捉局部细节的问题,提出将 U-Net 结构引入判别器,使其输出逐像素的判别结果,从而为生成器提供更丰富的梯度信号。项目基于 PyTorch,支持 BigGAN 和 FFHQ 等数据集,包含完整的训练和评估代码。核心能力包括:改进的判别器架构、稳定的训练流程、以及高质量的图像生成效果。该实现解决了 GAN 训练中模式崩溃和细节模糊的问题,尤其在高分辨率图像生成任务中表现突出。

开源 free 图像生成
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 403
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类图像生成
开发团队boschresearch
所属国家
定价模式free
价格说明开源项目,基于AGPL-3.0许可证,可自由使用和修改,无付费版本。
访问状态
是否开源是
开源协议AGPL-3.0
主要语言Python
技术栈/模型bcai,biggan,computer-vision,cvpr2020,ffhq,gan,generative-adversarial-network,image-generation,machine-learning,paper-resource,u-net,unet-gan
GitHub 星标★ 403
30天Star增速
HF 下载量
上线时间2020-08-27 00:00:00
最近更新2026-06-16 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2

使用教程

核心亮点

  • 判别器输出逐像素结果,为生成器提供更细粒度的梯度,显著改善图像局部纹理和边缘清晰度。
  • 官方实现,代码结构清晰,基于 PyTorch,易于复现论文实验,适合学术研究和二次开发。
  • 支持 BigGAN 和 FFHQ 等主流数据集,训练脚本完整,可直接用于高分辨率图像生成任务。

不足之处

  • 项目维护停滞,依赖较旧,可能不兼容新版 PyTorch,需自行适配。
  • 文档仅覆盖基础用法,缺少详细的参数调优指南和常见问题解答,上手门槛偏高。

适用场景

  • 学术研究:复现 CVPR 2020 论文,对比不同判别器架构对 GAN 性能的影响。
  • 高分辨率图像生成:在 FFHQ 等数据集上训练,提升人脸图像的细节真实感。
  • GAN 训练优化:作为判别器改进方案,集成到现有 GAN 流程中,缓解模式崩溃。

替代项目

StyleGAN2、SAGAN、Projected GAN

项目介绍

unetgan 是图像领域的开源项目,由 boschresearch 开发,2020 年首次发布。

它收录于图像分类,该分类目前共有 839 个项目,GitHub 星标数为 403。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-06-16。基于AGPL-3.0许可证,可自由使用和修改,无付费版本。

它主要面向的使用场景是:学术研究:复现CVPR2020论文,对比不同判别器架构对GAN性能的影响。同类可对比的替代方案包括 StyleGAN2、SAGAN、Projected GAN。

上一篇:RQ-Transformer

下一篇:StyleGAN3-CLIP-notebooks

同类项目推荐

satori 开源

网页代码一键变高清图,放大不糊,不用截图

Enlightened library to convert HTML and CSS to SVG

★ 13967 2026-08-09
C2GAN 开源

用关键点精准控制图像生成,让AI画出你想要的结构

[ACM MM 2019 Oral] Cycle In Cycle Generative Adversarial Networks for Keypoint-Guide···

★ 70 2026-08-09
ComfyUI 开源

像搭积木一样连节点,生图流程自己搭

The most powerful and modular diffusion model GUI, api and backend with a graph/node···

★ 134493 2026-08-09