StyleSwin

用 Transformer 生成高清图像,效果媲美 StyleGAN

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 52 🐛 待处理 13 👀 关注 7 ⚠️ 作者已归档(停止维护)
GitHub ↗
GitHub 星标 ★ 543
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

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

项目数据

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

项目介绍

StyleSwin 是一个基于 Transformer 的高分辨率图像生成模型,发表于 CVPR 2022。它针对传统 GAN 在生成高分辨率图像时依赖卷积结构、难以捕捉全局依赖的问题,将 Swin Transformer 引入生成器,结合风格调制机制,实现了从 256×256 到 1024×1024 的高质量图像合成。项目提供了完整的训练和测试代码,支持人脸、猫、教堂等数据集的生成,并公开了预训练权重。核心能力包括:多尺度 Transformer 生成器设计、与 StyleGAN 类似的风格控制、以及在高分辨率下保持全局一致性的能力。该项目为研究 Transformer 在生成任务中的潜力提供了重要参考,也验证了纯注意力机制在图像生成领域的可行性。

Star 增长趋势

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

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

使用教程

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

难度:进阶 约 60 分钟 部署方式:本地GPU 5 步

环境要求

  • PyTorch
  • CUDA 显卡
  • 人脸或通用图像数据集

安装与启动步骤

  1. 1拉仓库装依赖

    官方实现,按 requirements 装。

    python -m pip install -r requirements.txt
  2. 2备数据

    它按固定分辨率训练,数据要先对齐到同一尺寸。

  3. 3跑训练

    先小分辨率短轮次,确认能出图。

  4. 4看高频细节

    它的卖点是高分辨率下的纹理质量,重点看这块。

  5. 5对比同等算力下的卷积方案

    不对比就看不出 Transformer 结构到底值不值。

如何确认成功

小分辨率短轮次训练后能生成结构完整的图像,且放大看纹理比同参数量卷积基线更细

注意事项

  • 论文是 CVPR 2022,用 Transformer 做高分辨率 GAN 的代表工作之一。
  • 训练成本高,单卡短轮次只能验证管线通不通。
  • 对数据分辨率和裁剪方式敏感,预处理不对会明显掉画质。
  • 现在这类任务主流已转向扩散模型,它更适合当对照基线。

核心亮点

  • 首个将 Swin Transformer 用于高分辨率 GAN 生成,验证了纯注意力架构的可行性
  • 在 1024×1024 分辨率下生成质量接近 StyleGAN2,且全局结构更一致
  • 代码结构清晰,提供预训练模型和完整训练配置,易于复现

不足之处

  • 训练资源消耗大,普通 GPU 难以复现高分辨率实验
  • 推理速度较慢,实际应用部署成本高
  • 文档/社区待观察

适用场景

  • 学术研究:探索 Transformer 在生成对抗网络中的替代卷积方案
  • 高分辨率图像合成:生成人脸、动物、建筑等大尺寸逼真图像
  • 风格迁移与可控生成:利用风格调制实现不同属性的图像编辑

替代项目

StyleGAN2、TransGAN、SwinIR

上一篇:dream-factory

下一篇:CoCosNet-v2

同类项目推荐

stable-diffusion-nvidia-docker 开源

一键启动 GPU 版 Stable Diffusion,多卡并行出图快

GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web···

★ 369 2026-08-09
joliGEN 开源

用 GAN 和扩散模型搞定真实图像视频生成与编辑

Generative AI Image and Video Toolset with GANs and Diffusion for Real-World Applica···

★ 290 2026-08-09
ComfyUI-NO8D-controls 开源

精简 ComfyUI 工作流,提示词到出图一步到位

This is a set of high-efficiency custom nodes covering prompt writing to final image···

★ 317 2026-09-16
model-keyword 开源

选模型自动填触发词,省去手动复制粘贴

Automatic1111 WEBUI extension to autofill keyword for custom stable diffusion models···

★ 270 2026-09-28
discoart 开源

一行代码,用文本生成惊艳的 Disco Diffusion 艺术画作

Create Disco Diffusion artworks in one line

★ 3810 2026-08-09
opencode-gpt-imagegen 开源

在OpenCode里直接对话生成图片,省去切换工具

OpenCode plugin: image generation via your ChatGPT subscription

★ 121 2026-08-09
unpaint 开源

在 Windows 和 Xbox 上本地跑 Stable Diffusion,离线生成和修补图像

A simple Windows / Xbox app for generating AI images with Stable Diffusion.

★ 286 2026-08-09
ArtGAN 开源

用 GAN 生成指定风格的艺术画作,让 AI 当画家。

ArtGAN + WikiArt: This work presents a series of new approaches to improve GAN for c···

★ 432 2026-08-09