当前位置:   article > 正文

【深度学习】【Lora训练2】StabelDiffusion,Lora训练过程,秋叶包,Linux,SDXL Lora训练_秋叶 ai 训练lora

秋叶 ai 训练lora

上篇文谈论了一些基础使用,接下来实验一些更细节的问题。

文章链接:

kohya_ss:

https://qq742971636.blog.csdn.net/article/details/138135484

秋叶包基础:

https://qq742971636.blog.csdn.net/article/details/138195344

一、如何为图片打标

1.1. 打标工具

1.1.1. 秋叶中使用的WD1.4

一种词汇打标器,原始训练数据应该是来源于Danbooru 图像, Danbooru 就类似于分类,表述图中有的元素单词,这种倾向于是一个一个的单词(tag)来打标图片。

在这里插入图片描述
图片打标结果展示如下,一行是一个图片的打标:

solo, monochrome, 1girl, portrait, greyscale, short hair, realistic, traditional media, lips, smile

no humans, monochrome, greyscale, traditional media
  • 1
  • 2
  • 3

1.1.2. 使用BLIP2

BLIP2,这个在kohya_ss中有可视化界面,或者使用程序:

https://huggingface.co/Salesforce/blip2-opt-2.7b

BLIP2的提示语更倾向于是一个短语短句子。

1.1.3. 用哪一种

都差不多,最终都会被CLIP转为嵌入量,只要打标准确达意就好。

二、 Lora训练数据的要求

参考资料:https://zhuanlan.zhihu.com/p/676456908

2.1 图片要求

数量我觉得质量越好、数量越多肯定是最好的。

分辨率适中,勿收集极小图像。

数据集需要统一的主题和风格的内容,图片不宜有复杂背景以及其他无关人物。

图像人物尽量多角度,多表情,多姿势。

凸显面部的图像数量比例稍微大点,全身照的图片数量比例稍微小点。

堆糖:https://www.duitang.com
花瓣:https://huaban.com
pinterest:https://www.pinterest.com

通常,准备数百张图像是理想的(图像数量太少会导致类别图像无法被归纳,特征也不会被学习)。

如果要使用生成的图像,生成图像的大小通常应与训练分辨率(更准确地说,是bucket的分辨率,见下文)相匹配。

2.2 图片的打标要求

如果想要用文字自由控制头发是什么颜色,那么tag中对头发颜色的描述就可以保留。

如果是某个特定的人物ID或者画风,不想要文字控制,想要Lora直接就有效果,那么就需要删除这种描述文字。

在这里插入图片描述

三、 Lora的其他问题

qa1

为什么lora有附加提示词?:如之前的概念,附加提示词是为了更好触发Lora功能,如果在后续Lora使用中,加上这个附加提示词去生图,那么就嘎嘎被控出Lora风格,附加提示词最好整点与众不同的词。

qa2

lora训练多少轮合适?:10轮,20轮。看到还有200轮的,loss收敛就好。

qa3

Lora模型的不同之处:网络结构(LoRA/LoCon/LoHa/DyLoRA)?:详细看秋叶包的解释。

qa4

Lora的正则化数据,这种训练数据可以用模型生成或者自己找,比如画风Lora训练中,只想生成卡通人物,那么正则数据就可以选真实人物。

qa5

你为什么会觉得BLIP2的标记更好?

WD1.4的打标是一些tag词,比如我训练的素描风格数据,WD1.4给的tag词里有很多相同的词,比如“monochrome, greyscale, traditional media ”,这些词导致了在后续我使用Lora生图过程中,我需要加这种描述词才能很好地触发Lora风格。而BLIP2的标记只会描述物体内容,就不至于出现这种问题。

只能说,各有长处,每个标记txt文件或许要审查一下更好,偷懒地话用BLIP2或许可以更懒一点。

在这里插入图片描述

四、 对图片的处理细节

比如我要训练SDXL的Lora,这种Lora可以将图片改为素描风格,我需要什么样的图片?我应该如何处理图片?哪种图片适合?

4.1. 图片尺寸问题

我需要的是高清图,要有足够的细节。SDXL本身是1024*1024的适应,所以我找的图不能太小,最小也应该有个768的大小。这一点上,可以借助一些超分算法和美化算法来调整图片尺寸,让图片有足够多的细节。故图片最小边大于512就足够好了,不用苛刻太多。

4.2. 图片内容选取问题

基本概念是,模型学习的是一种映射关系,对于没见过怎么转换的,迁移能力不是很强。比如我训练了中国墨水化的Lora,都是用的一些风景图片训练的,那么用这个Lora去生成人物是很垃圾的。模型见过类似的图片映射,才能聪明起来,这是训练的核心。

基于此,如果我想要的是画风Lora,那么我搞的数据应该尽量是各种各样的数据都来点;如果我想要的是某个人物的样貌Lora,那么最好搞的数据就全是这个人物的图片。

在素描风格中,我找了一些乱七八糟的图,我希望Lora学习到的是一种笔触风格:
在这里插入图片描述

4.3. 什么是一张合适的图?

基本原则是,凡是带有干扰的图,我都不会要,我都会修改。我要纯粹表达Lora的意向,图片要纯粹

4.3.1. 解决水印问题——inpainting

下图的红圈内都是不想要的图片内容,需要想办法去除:

在这里插入图片描述
win10企业版本的图片查看器可以一定程度修改这种情况:

在这里插入图片描述
可以看到擦出了很多,有人问下图还有擦不掉的怎么做,当然是直接把这张图扔垃圾桶里。

在这里插入图片描述

4.3.2. 解决边界的问题——裁剪

人物或者画风的Lora训练都需要遵循“图片要纯粹”。

比如画风中这张图的边界其实是一个很差的点,我们绝对不想要Lora指导生图出来的图中有这种难看的边界:

在这里插入图片描述
裁剪掉即可:

在这里插入图片描述

人物Lora训练也有类似的情况,比如这张图,如果是人物Lora,就需要裁剪这张图,因为要突出纯粹的核心内容。

在这里插入图片描述

裁剪后是这样的图才行,甚至不想绘制手,也可以把下半身裁剪一些,毕竟Lora那么傻,这么复杂的腿部姿势构图它可能学不会。

在这里插入图片描述

4.3.3. 解决主体多的问题——删除

训练素描风格,下面这个图不合适,主体太多,提示词tag很难形成映射。类似地人物Lora也是一样的道理,不要乱七八糟的无关人物干扰最好,否则就删除那张图。
在这里插入图片描述

4.3.4. 解决审美的问题——删除

训练素描风格,下面这个图不合适,需要删除。原因在于太丑,我不希望Lora指导绘图绘制出来这么拙略的出图。Lora只是一个小孩子,学映射没那么聪明,这种干扰不要当训练数据给进去,不然Lora学不好。

人物Lora也是类似地,人物如果丑,角度不好,就尽量不要用来当训练数据。

在这里插入图片描述

问询、帮助

你如果需要帮助,请看这里:

https://docs.qq.com/sheet/DUEdqZ2lmbmR6UVdU?tab=BB08J2
  • 1
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/617890
推荐阅读
相关标签
  

闽ICP备14008679号