StyleKeeper

用负向引导,让风格迁移不偷内容

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

项目预览

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

项目数据

分类图像生成
开发团队naver-ai
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0 许可证,完全免费,需自行部署。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型diffusion,stable-diffusion
GitHub 星标★ 482
上线时间2024-02-20 00:00:00
代码更新2025-12-02
仓库体积37.7 MB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

StyleKeeper 是一个基于 PyTorch 的官方实现,针对扩散模型(如 Stable Diffusion)在风格迁移或风格化生成时,内容信息从参考图像泄漏到生成结果中的问题。它通过引入“负视觉查询引导”(Negative Visual Query Guidance)机制,在生成过程中主动抑制与参考图像内容相关的特征,从而在保留风格的同时防止内容泄露。核心能力包括:提供完整的训练和推理代码、支持自定义风格参考、与主流扩散模型框架兼容。该项目解决了风格迁移中风格与内容解耦的难题,提升了生成图像的内容独立性。

Star 增长趋势

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

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

使用教程

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

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

环境要求

  • Python
  • CUDA GPU
  • xformers

安装与启动步骤

  1. 1装依赖

    它把一长串包锁在一行里,版本基本不能动。

    pip install --upgrade diffusers accelerate transformers einops kornia gradio triton xformers==0.0.16
  2. 2跑文生图风格化

    先用自带示例风格确认环境通了。

    python vsp_script.py --style fire
  3. 3加控制条件

    边缘与深度两种条件各有一个脚本。

    python vsp_control-edge_script.py --style fire --controlnet_scale 0.5 --canny_img_path assets/edge_dir
  4. 4迁移真实图片

    换成真实照片走另一条路径。

    python vsp_real_script.py --img_path assets/real_dir --tar_obj cat --output_num 5
  5. 5看注意力图

    理解它凭什么保持风格,就看这一步的结果。

    python visualize_attention_src/save_attn_map_script.py

如何确认成功

用自带示例风格跑一遍文生图脚本,能输出风格化结果,且注意力可视化脚本能生成对应的热力图

注意事项

  • 它是论文配套代码,研究风格保持与注意力注入机制,不是开箱即用的美化工具。
  • xformers 锁在 0.0.16,新环境里这一步最容易失败,装不上先跑纯 PyTorch 分支。
  • 三个脚本分别对应文生图、控制条件、真实图迁移,参数不能混着填。
  • 显存需求随输出分辨率和批量增大,先用小图跑通。

核心亮点

  • 提出新颖的负视觉查询引导机制,有效解决内容泄漏问题
  • 官方 PyTorch 实现,代码结构清晰,便于二次开发
  • 基于 Stable Diffusion,生态兼容性好,易集成到现有流程

不足之处

  • 项目处于早期阶段,文档和社区支持待观察
  • 依赖特定扩散模型,泛化性需进一步验证

适用场景

  • 艺术风格迁移,需保持内容独立性
  • 图像编辑,避免参考图内容干扰
  • 创意生成,需精细控制风格与内容分离

替代项目

StyleGAN、CycleGAN、DiffusionArt

同类项目推荐

grok-image-api 开源

几行代码调通 Grok 文生图与改图,价格和审核规则一目了然

Grok Image API (Grok Imagine 1.5, grok-imagine-1.5-apimart): model ids, per-image pr···

★ 145 2026-09-27
cog-consistent-character 开源

上传一张角色图,生成各种姿势但脸不崩的图

Create images of a given character in different poses

★ 725 2026-08-09
AI-Visual-Prompt-Cookbook 开源

复制 JSON 模板,秒出高质量 AI 图像

Plug-and-play JSON style prompts for Nano Banana Pro, GPT Image & Midjourney. Copy o···

★ 838 2026-08-09
awesome-aiart-pics-prompts 开源

3000+ 精选提示词,让 AI 出图更精准

精选 3000+ Gemini Nano Banana Pro 高质量提示词与生成案例 | 涵盖摄影、设计、艺术、营销···

★ 545 2026-08-09
ImageLens 开源

看中网页图片,一键反推提示词并立即生成测试效果

一个可以帮你快速反推出网页上图片的提示词,并快速生图测试提示词效果的插件。A Chrome e···

★ 191 2026-08-09
dynamicprompts 开源

用模板批量生成多样化提示词,告别手写重复

Templating language for generating prompts for text to image generators such as Stab···

★ 167 2026-09-17
openfill 开源

在 GIMP 里框选即用 AI 填充、去物、扩图,全本地跑

Generative Fill, Remove object, Generative Expand and Text to Image for PhotoCraft a···

★ 248 2026-11-07
zenith-image-generator 开源

一个接口聚合多家AI画图,一键部署到边缘

Modern AI image generator with multi-provider support (Gitee AI, HuggingFace, ModelS···

★ 225 2026-08-09