soft-intro-vae-pytorch

用软对抗训练让VAE生成更清晰图像，训练更稳定

开源 🏷️完全免费 🔧 仅开源项目，无在线服务，需自行部署 图像生成 🍴 复刻 31 🐛 待处理 2 👀 关注 7
GitHub ↗
GitHub 星标 ★ 197
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目，基于Apache-2.0许可证，可免费使用和修改，无付费版本。

项目预览

soft-intro-vae-pytorch 项目界面预览图
图片来源：GitHub 官方仓库名片

项目数据

分类图像生成
开发团队taldatech
GitHubhttps://github.com/taldatech/soft-intro-vae-pytorch
定价模式🏷️完全免费
价格说明开源项目，基于Apache-2.0许可证，可免费使用和修改，无付费版本。
访问状态🔧 仅开源项目，无在线服务，需自行部署
是否开源是
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型cvpr2021,density-estimation,image-generation,pytorch,soft-intro-vae,soft-introvae,vae,vae-pytorch,variational-autoencoder
GitHub 星标★ 197
上线时间2020-12-15 00:00:00
代码更新2022-06-27
仓库体积1.3 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

Soft-IntroVAE 是 CVPR 2021 Oral 论文的官方 PyTorch 实现，旨在改进内省变分自编码器（Introspective VAE）的训练稳定性和生成质量。传统 VAE 在生成高分辨率图像时容易产生模糊，而 GAN 虽清晰但训练不稳定。该项目通过引入软化的对抗训练目标，让编码器和解码器在训练中相互博弈，同时保持 VAE 的概率建模优势，从而在密度估计和图像生成任务上取得更优表现。核心能力包括：支持多种数据集（如 MNIST、CIFAR-10、CelebA）的训练与评估，提供预训练模型和生成样本可视化，代码结构清晰，便于二次开发。项目解决了 VAE 生成模糊和 GAN 训练不稳定的问题，为生成模型研究提供了一个稳健的基线。

Star 增长趋势

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

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

核心亮点

  • 官方实现，代码质量高，结构清晰，易于复现论文结果
  • 训练稳定性优于传统VAE，生成图像更清晰，密度估计更准确
  • 支持多数据集和预训练模型，开箱即用，适合快速实验

不足之处

  • 文档/社区待观察
  • 计算资源需求较高，训练大模型时对显存要求严格

适用场景

  • 学术研究：作为VAE改进基线，对比生成质量与密度估计
  • 图像生成：生成高分辨率、清晰的自然图像
  • 教学演示：理解内省变分自编码器与对抗训练的结合

替代项目

NVAE、VQ-VAE-2、IntroVAE

上一篇：StoryGAN

下一篇：Conditional-GAN

同类项目推荐

Image-MetaHub 开源

本地管理AI生成图，一键回传ComfyUI复现

Local-first AI image organizer and generative media library manager for ComfyUI, A11···

★ 335 2026-08-09
NFTgenerator 开源

拖拽图层，一键生成上千个NFT变体。

SickAlien NFT Art Generator is an interactive tool to generate 1000s of NFTs just by···

★ 131 2026-08-09
clarity-upscaler 开源

糊图一键变高清，免费开源随便用

Clarity AI | AI Image Upscaler & Enhancer - free and open-source Magnific Alternativ···

★ 5172 2026-08-09
text-to-image 开源

用一句话文字描述，自动生成对应图像

Generative Adversarial Text to Image Synthesis / Please Star -->

★ 599 2026-08-09
AdaIN-style 开源

上传任意风格图，实时迁移到内容图，无需重新训练

Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization

★ 1573 2026-08-09
Boogu-Image 开源

少一个数量级数据，也能做出接近闭源效果的图像生成与编辑模型

Boogu-Image-0.1 is an Apache-2.0 open-source image generation and editing model fami···

★ 999 2026-11-06
invertible-cd 开源

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

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

★ 100 2026-08-09