anycost-gan

按需选择计算成本,让 GAN 在普通设备上也能实时玩转图像编辑

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 98 🐛 待处理 6 👀 关注 20
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 779
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

anycost-gan 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队mit-han-lab
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型computer-graphics,computer-vision,deep-learning,gan,gans,generative-adversarial-network,image-editing,image-generation,image-manipulation,pytorch,stylegan2
GitHub 星标★ 779
上线时间2021-03-04 00:00:00
代码更新2023-10-03
仓库体积17.1 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

Anycost GANs 是 CVPR 2021 收录的交互式图像合成与编辑框架,由 MIT 等机构提出。它解决的是 GAN 模型在推理时计算开销大、难以在普通设备上实时交互的问题。核心思想是训练一个支持弹性计算的生成器,用户可以根据硬件资源或延迟需求,动态选择不同的子网络(从轻量到完整)进行推理,而无需重新训练。项目提供了基于 StyleGAN2 的预训练模型,支持图像生成、属性编辑、潜空间插值等任务,并包含用于人脸编辑的编码器。其亮点在于实现了‘任意成本’的推理,即在不同计算预算下都能获得质量可接受的输出,同时通过子网络间的协同训练减少了质量损失。该代码库基于 PyTorch,提供了训练、评估和交互式演示的完整流程,适合研究弹性 GAN 和部署资源受限的生成应用。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 779

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:进阶 约 40 分钟 部署方式:本地安装 4 步

环境要求

  • NVIDIA GPU 11G+
  • Python 3.8+

安装与启动步骤

  1. 1克隆

    可变计算预算的 GAN,同一模型可快可慢。

    git clone https://github.com/mit-han-lab/anycost-gan.git
    cd anycost-gan
  2. 2装依赖

    注意 CUDA 算子编译。

    pip install -r requirements.txt
  3. 3跑示例

    原生实现可能比预期慢,可强制原生模式。

    python demo.py
  4. 4算指标

    FID 与属性一致性是它的两个关键指标。

    python metrics/fid.py --path ./checkpoints

如何确认成功

不同预算档位下出图,画质随预算单调变化。

注意事项

  • 它的价值是「一个模型覆盖多档算力」,适合端侧弹性部署。
  • FORCE_NATIVE=1 可绕过自定义算子,方便调试。
  • 多卡训练用 horovodrun,单卡别用。

核心亮点

  • 弹性推理机制:可根据硬件条件动态切换子网络,从轻量到完整,灵活控制延迟和显存占用
  • 质量损失小:通过子网络协同训练,在 2-4 倍加速下仍能保持接近完整模型的图像质量
  • 附带交互式演示:提供 Web 界面,支持实时图像编辑,直观展示不同成本下的效果

不足之处

  • 文档/社区待观察
  • 预训练模型仅覆盖人脸等特定领域,泛化到其他类别需自行训练

适用场景

  • 在低端 GPU 或 CPU 上实现实时图像编辑交互
  • 移动端或边缘设备部署生成式 AI 应用
  • 研究 GAN 模型压缩与自适应推理策略

替代项目

GAN Compression、Conditional GANs (cGANs)、StyleGAN2

上一篇:scene_generation

下一篇:collage-diffusion-ui

同类项目推荐

StabilityMatrix 开源

一站式管理 Stable Diffusion 环境与模型,告别命令行折腾

Multi-Platform Package Manager for Stable Diffusion

★ 9000 2026-08-09
easydiffusion 开源

一键安装,浏览器里玩转 AI 绘画,零基础也能出图

An easy 1-click way to create beautiful artwork on your PC using AI, with no tech kn···

★ 10474 2026-08-09
grok-imagine-api 开源

照着提示词配方和轮询示例,快速接入 Grok 图像生成与编辑

Grok Imagine API guide: image generation and editing on the grok-imagine routes, per···

★ 134 2026-10-06
Imaginepy 开源

零成本调用 Midjourney 级图像生成,无需付费 API。

ImaginePy, opensource (reversed) API like Stable Diffusion, Midjourney. Everything a···

★ 84 2026-08-09
openjourney 开源

用 Gemini 打造你自己的 MidJourney,生成图像和视频

Open-source clone of the MidJourney web interface featuring real AI image and video ···

★ 272 2026-08-09
invertible-cd 开源

7步内完成高质量文本引导图像编辑,让扩散模型编辑提速十倍。

[NeurIPS'2024] Invertible Consistency Distillation for Text-Guided Image Editing in ···

★ 100 2026-08-09
aperture 开源

一眼看透 Stable Diffusion 在关注什么,调试生成更轻松。

Visualize attention layers from Stable Diffusion

★ 91 2026-08-09
ai-image-clean-eraser 开源

上传图片,AI 自动找字手动擦,秒出高清无痕图

AI-Powered Text Remover 2026: Auto-Detect & Manual Precision with HD Quality

★ 119 2026-09-30