dreambooth-keras

用 TensorFlow 微调 Stable Diffusion,实现个性化图像生成。

dreambooth-keras 是一个基于 KerasCV 和 TensorFlow 的 DreamBooth 实现,旨在让开发者无需依赖 PyTorch 生态即可进行个性化文本到图像模型的微调。它解决了 DreamBooth 原版实现仅支持 PyTorch 的问题,为 TensorFlow 用户提供了完整的训练和推理流程。项目核心能力包括:支持从少量用户提供的图像(通常 3-5 张)中学习特定对象或风格,并将该概念绑定到唯一标识符(如 'sks')上,从而在生成新图像时保持对象一致性。代码以 Jupyter Notebook 形式提供,易于上手和修改,适合希望深入理解 DreamBooth 原理或将其集成到 TensorFlow 工作流的开发者。项目处于早期阶段,功能相对基础,但为 TensorFlow 社区填补了空白。

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

项目数据

分类图像生成
开发团队sayakpaul
所属国家
定价模式free
价格说明开源项目,Apache-2.0 许可证,完全免费,可自行部署使用。
访问状态
是否开源
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型computer-vision,dreambooth,generative-ai,keras,keras-cv,stable-diffusion,tensorflow
GitHub 星标★ 88
30天Star增速
HF 下载量
上线时间2023-01-15 00:00:00
最近更新2025-03-17 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数4

使用教程

核心亮点

  • 基于 KerasCV,与 TensorFlow 生态无缝集成,便于复用现有 TF 工具链
  • 以 Notebook 形式提供,步骤清晰,适合学习和快速实验
  • 填补了 DreamBooth 在 TensorFlow 生态的空白,为 TF 用户提供替代方案

不足之处

  • 项目处于早期阶段,功能较基础,可能缺少高级优化(如低秩适应)
  • 文档和社区支持有限,遇到问题可参考资源较少

适用场景

  • 在 TensorFlow 环境中微调 Stable Diffusion 模型,实现个性化生成
  • 研究 DreamBooth 原理,通过可读代码学习训练流程
  • 作为基础,集成到现有 TF 图像生成应用中

替代项目

diffusers (Hugging Face DreamBooth 实现)、DreamBooth (原版 PyTorch 实现)、SimpleTuner (基于 PyTorch 的微调工具)

项目介绍

dreambooth-keras 是模型训练领域的开源项目,由 sayakpaul 开发,2023 年首次发布。

它收录于模型训练分类,该分类目前共有 518 个项目,GitHub 星标数为 88。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2025-03-17。Apache-2.0 许可证,完全免费,可自行部署使用。

它主要面向的使用场景是:在TensorFlow环境中微调StableDiffusion模型,实现个性化生成。同类可对比的替代方案包括 diffusers (Hugging Face DreamBooth 实现)、DreamBooth (原版 PyTorch 实现)、SimpleTuner (基…。

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13