File size: 2,602 Bytes
9469607
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
---
license: apache-2.0
---

# Krea 2 深度参考 LoRA(Depth LoRA)

## 简介

这是一个Krea 2 深度参考 LoRA,可以将深度图转换为高质量的图像。模型在多种数据集上分阶段训练,兼顾整体结构准确性与细节还原能力。

## 训练细节

训练采用两阶段、分辨率递进的策略:

- **第一阶段 —— 512 分辨率预训练**
  使用从公开深度数据集中采样的 2000 对图像子集进行训练,所有标注均通过 VLM 重新生成(VLM re-caption),以提升图文对齐质量。该阶段使用恒定学习率 `1e-4`,建立扎实的深度结构先验。

- **第二阶段 —— 1536 高分辨率精修**
  在 1536 分辨率的深度数据集上继续训练,重点提升细节重建能力(边缘、细小结构与精细几何)。该阶段改用余弦(cosine)学习率调度,使收敛更加稳定。

| 项目 | 参数 |
| --- | --- |
| LoRA Rank | xxx |
| 第一阶段 | 512 分辨率,2000 对图像(公开深度数据集 + VLM 重打标),恒定学习率 `1e-4` |
| 第二阶段 | 1536 分辨率深度数据集,余弦学习率调度 |
| 总训练轮数 | 50+ epoch |

## 使用方法

⚠️ **本 LoRA 必须配合 EditUtils 插件使用:**

👉 ComfyUI-EditUtils

1. 安装 ComfyUI-EditUtils 插件。
2. 在工作流中加载本 LoRA。
3. 输入原始图像,即可得到对应的深度图。

## 注意事项

- 建议输入图像最长边在 1024 ~ 2048 之间;输入分辨率越高,深度细节越丰富。512 分辨率可能无法正常工作。
- 若不配合 EditUtils 插件使用,将无法正常发挥效果。

## 📬 联系方式

- 𝕏 **Twitter**: [@Lrzjason]
- 📧 **邮箱**: [lrzjason@gmail.com]
- 💬 **QQ 群**: 866612947
- 💬 **微信号**: fkdeai
- 🎨 **CivitAI**: [xiaozhijason]

## 示例

![ComfyUI_temp_aelyb_00001_flbfp_1785244953](https://cdn-uploads.huggingface.co/production/uploads/63891deed68e37abd59e883f/7ga9DifCzdD3ymfVUc1mX.jpeg)

![ComfyUI_temp_prnep_00001_ukehj_1785244125](https://cdn-uploads.huggingface.co/production/uploads/63891deed68e37abd59e883f/kCrmMM58d9O-EoiAsml77.jpeg)

![ComfyUI_temp_yueag_00002_djzyk_1785244612](https://cdn-uploads.huggingface.co/production/uploads/63891deed68e37abd59e883f/O3rRSw2Xdie7ffhFy-VUg.jpeg)

![ComfyUI_temp_dahts_00001_ccpdc_1785263076](https://cdn-uploads.huggingface.co/production/uploads/63891deed68e37abd59e883f/rStu59gXHlJouNDX2hsWr.jpeg)

![ComfyUI_temp_dahts_00003_nghhr_1785263436](https://cdn-uploads.huggingface.co/production/uploads/63891deed68e37abd59e883f/SgxuS_Vpz-vSV_GEEZiU_.jpeg)