Skip to content

Commit 12bfbb6

Browse files
committed
update
1 parent 3f97d11 commit 12bfbb6

File tree

3 files changed

+41
-0
lines changed

3 files changed

+41
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
---
2+
title: 开源生图模型 stable diffusion lora 训练记录
3+
categories: [模型训练]
4+
description: 帮自媒体朋友做的生图模型训练研究
5+
keywords:
6+
- stable diffusion
7+
- 扩散模型
8+
- lora 训练
9+
date: 2023-08-16
10+
draft: false
11+
---
12+
13+
### 场景
14+
15+
- 自媒体运营,使用生成的穿搭图、风景图之类的图片引流
16+
17+
### 数据准备
18+
19+
- 穿搭图、旅游图:小红书下载穿搭、旅游博主的图片,本文示例图15张
20+
- 抽象类设计图:设计师网站下载图片
21+
22+
### 训练流程
23+
24+
- 底模选择:sd 真人类型底模麦橘写实
25+
- 训练脚本:kohya_ss
26+
27+
#### 数据预处理与参数
28+
29+
- 基本数据:训练图片15张图
30+
- 数据增强:upscaler 到 2k,yolo 抠出人体,paddle matting 抠出更细致的人体轮廓,除去人体外的像素换成白底,左右翻转加强
31+
- 训练硬件:显卡2080Ti
32+
- 训练参数:network_alpha 32,network_dim 32,batchsize 8,重复次数6,训练轮次20轮
33+
34+
#### 结果分析
35+
36+
- 使用 xyz plot 插件查看训练轮次-lora强度的对比图:
37+
- ![xyz plot](https://github.com/luokai0223/luokai0223.github.io/raw/master/pics/2023-08-16-stable_diffusion_lora/2.png)
38+
- 本次测试的真人眼间距有一点点宽,中庭偏长,额头有点大。在 lora 强度较大时能明显发现五官缺陷被放大。
39+
- 图片不宜过多,试过100多张图片,画面光影、背景、人物眼神异常,原始人物五官缺陷被放大。
40+
- 训练轮次2轮内,强度设置0.7也能看见原始人物脸部特征,但是眼神和光影异常。
41+
- 总结:15张图左右训练20轮次以内即可以完成,使用强度设置为0.5-0.7观感比较好,既有人物特点,也没有放大缺陷。
12.5 KB
Loading
1.57 MB
Loading

0 commit comments

Comments
 (0)