dreambooth-keras

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

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

项目预览

dreambooth-keras 项目界面预览图
图片来源:项目官网

项目数据

分类图像生成
开发团队sayakpaul
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0 许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型computer-vision,dreambooth,generative-ai,keras,keras-cv,stable-diffusion,tensorflow
GitHub 星标★ 88
上线时间2023-01-15 00:00:00
代码更新2023-02-16
仓库体积409 KB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数5

项目介绍

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

Star 增长趋势

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

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

核心亮点

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

不足之处

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

适用场景

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

替代项目

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

同类项目推荐

lora 开源

用少量图片和普通显卡,快速训练专属图像模型

Using Low-rank adaptation to quickly fine-tune diffusion models.

★ 7558 2026-08-09
litData 开源

几行代码优化数据管道,让 GPU 不再等数据

Speed up model training by fixing data loading.

★ 617 2026-09-19
AdaptiveDiffusion 开源

不重训直接加速扩散模型,生成更快更省算力

[NeurIPS'24] Training-Free Adaptive Diffusion with Bounded Difference Approximation ···

★ 74 2026-08-09
sdnq 开源

低比特量化扩散模型,小显存也能跑图

SD.Next Quantization Engine

★ 139 2026-10-17
verl-omni 开源

给扩散和全模态模型做 RL 后训练,一套框架跑通

Multimodal RL training framework for diffusion & omni models

★ 1472 2026-10-12
Rearchitecting-LLMs 开源

手把手教你剪枝、蒸馏、优化注意力,免费GPU跑通

Official code for the Manning book on structural LLM optimization: depth/width pruni···

★ 79 2026-08-09
Text-To-Video-Finetuning 开源

用 Diffusers 微调文本生成视频模型,定制专属视频风格

Finetune ModelScope's Text To Video model using Diffusers

★ 705 2026-08-09
STEPQuant 开源

定位循环状态量化误差,让低比特 LLM 更稳

STEPQuant: When and Where Errors Matter in Delta-Rule Recurrent State Quantization

★ 102 2026-10-31
Automodel 开源

Hugging Face 模型直接多卡分布式微调,不改代码。

Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face sup···

★ 1079 2026-09-16