基于开源Stable Diffusion大模型训练Lora绘制像素风动漫图
空怯独爱控
2024年08月13日 19:54

一、引言

大家好呀!今天想和大家分享一下如何基于开源的 Stable Diffusion 大模型训练 Lora 来绘制超可爱的像素风动漫图。

cut-off

二、Stable Diffusion 大模型

Stable Diffusion 是一款强大的深度学习模型,能够根据输入的提示和条件生成逼真的图像。它的出现极大地改变了图像生成的领域,为艺术家和创作者提供了新的工具和灵感。

推荐几个好用的模型网站:

可以了解自己喜欢的模型(合适的底模是训练属于自己的Lora的基础)

https://civitai.com/(需要科学上网)

https://huggingface.co/(模型资源多,不用科学上网,下载速度快)

stable-diffuson官模地址(一定要先上手会使用官模)

https://huggingface.co/CompVis/stable-diffusion-v1-4

cut-off

三、Lora 训练的原理与部分操作流程

Lora(Low-Rank Adaptation)是一种轻量级的模型训练方法,可以在不改变原始大模型的基础上,针对特定的任务或风格进行微调。在绘制像素风动漫图方面,Lora 能够快速捕捉到像素风格的特征,提高生成图像的质量和风格一致性。

1.数据准备

  • 收集大量与您想要训练的主题或风格相关的图像数据。建议使用至少几百张图像素材来获得较好的效果,但具体数量取决于多个因素。我这边训练的是像素风Lora,使用到了一百张像素动漫角色图。

  • 对数据进行清洗、筛选和预处理,例如调整图像大小、裁剪、归一化等。检查图片类型符合像素风格,推荐使用统一尺寸的图进行训练,可以提高训练效率。

  • 打标操作,对收集到的图片数据集统一进行打标,可以通过stable-diffusion插件选择“数据集标签编辑器”进行打标,打标时需要进行关键词排查,删去不合理的关键词,同时统一添加pixel art(像素艺术)标签。

2.环境搭建和基础模型选择

  • 安装所需的深度学习框架,如 PyTorch。因为只是素材不多的Lora训练,我这边选择Colab配合谷歌硬盘Drive进行Lora训练(Colab新用户可白嫖一段时间的Gpu使用,提高训练效率。专业模型训练还是推荐使用autodl快速部署训练)

  • 打包好的可视化训练文档链接:

https://colab.research.google.com/github/WSH032/lora-scripts/blob/main/Colab_Lora_train.ipynb

  • 具体流程不多说,参考下面教程

【AI绘画】零要求!傻瓜式!有网即可!免费云平台部署方式在Colab训练你的LoRA模型​

底模选择novelailatest-pruned----基于 NovelAI 相关技术的一个经过剪枝处理的模型。

训练参数可以推荐看秋葉aaaki​青龙圣者​两位大佬视频学习配置,需要根据训练目标和效果调整。不同的损失函数参数可能相互影响,因此需要尝试同时调整多个参数的组合,以找到最优的平衡。由于像素风训练侧重风格的一致性或特定风格的生成,我们可以稍微提高风格损失的权重。

由于训练过程长,需要大量时间,建议在训练过程中采样以实时查看训练效果,方便后续选择最满意的训练模型。

cut-off

四、成果展示

选取了2024年动漫女角色排名前一百并进行像素化批处理(由于是批处理操作没有对每张图片单独打角色标签,没有针对具体角色进行具体Lora训练,因此生成的像素图更多受到底模影响,着重关注图片形态特征,不保留角色样貌特征)

素材:

结果: