Stablediffusion拓展之dreambooth训练

时间:23-10-02 14:07:20

作者:

分类:AI软件教程

  大家好,我是电脑系统之家的小编。今天要为大家介绍一款名为DreamBooth的拓展训练功能。DreamBooth是一个基于Stable Diffusion的AI图像生成工具,它能够个性化定制输出图片,并在AI图像生成领域引起了广泛关注。

    7529d650b9fdf7747e71b286ddbd3e60

  什么是DreamBooth

  稳定扩散(Stable Diffusion)原本无法进行稳定的对象输出,每次生成的结果都是随机的,没有相关性。但是,通过引入Google的一项研究成果,DreamBooth解决了这个问题。DreamBooth在原始网络的基础上引入了Prior Preservation Loss,通过简单的训练,可以重新采样潜空间,实现指定对象的输出。这种个性化定制输出效果使得DreamBooth成为了AI图像生成领域最受欢迎的功能之一。

  DreamBooth的基本原理

  原始的DreamBooth是基于Google的Imagen和T5网络的,但是有开发者利用Stable Diffusion 和 CLIPText重新实现了一遍,这也是我们目前使用的DreamBooth版本。DreamBooth需要预先进行训练才能使用,但现在已经提供了用户界面(UI),不再需要手动输入命令来训练。

  安装DreamBooth扩展

  在启动Stable Diffusion的WebUI后,点击最右边的”Extension”选项卡,然后选择”Available”子选项卡,再点击”Load From”按钮。这时会显示一系列可用的扩展列表,选择”Install”按钮来安装DreamBooth(安装过程的日志可以在命令窗口中查看)。

  DreamBooth扩展插件

  值得一提的是,这个插件有些问题,虽然可以使用,但存在一些不完善之处。对于有经验的用户,我建议在requirements.txt文件中注释掉torch、torchvision、numpy、gradio和transformers等重要库的安装,而使用Stable Diffusion的库。

  进一步说,一些用户可能会遇到torch的覆盖性安装问题。这种情况下,最好在install.py文件中注释掉两行代码,并自行安装torch,因为这样会更方便。

  以上就是关于DreamBooth拓展训练的介绍。通过使用DreamBooth,我们可以实现个性化定制的图像生成,这在AI图像生成领域中非常受欢迎。如果你的硬件配置不满足DreamBooth功能的显存要求,你可以考虑使用云计算的方法,比如谷歌Colab云计算入口。希望这篇文章对你有所帮助!如果你对其他内容还有疑问,欢迎随时与我们交流。谢谢大家的阅读!