Skip to content

Latest commit

 

History

History
207 lines (163 loc) · 10.1 KB

README_zh-CN.md

File metadata and controls

207 lines (163 loc) · 10.1 KB

EasyPhoto | 您的智能 AI 照片生成器。

🦜 EasyPhoto是一款工具，用于生成AI肖像画，该代码可用于训练与您相关的数字分身。

🦜 🦜 Welcome!

English | 简体中文

目录

简介

EasyPhoto是一款工具，用于生成AI肖像画，该代码可用于训练与您相关的数字分身。建议使用 5 到 20 张肖像图片进行训练，最好是半身照片且不要佩戴眼镜（少量可以接受）。训练完成后，我们可以在推理部分生成图像。我们支持使用预设模板图片与上传自己的图片进行推理。

请阅读我们的开发者公约，共建美好社区 covenant | 简体中文

新特性：

  • 我们同样支持Stable Diffusion Webui插件. [🔥 2023.09.19]
  • 创建代码！现在支持 Windows 和 Linux，支持对背景进行微调，支持不同预测基础模型，支持多人生成。[🔥 2023.09.19]

这些是我们的生成结果: results_1 results_2 results_3

我们的ui界面如下:
训练部分: train_ui 预测部分: infer_ui

TODO List

  • 支持中文界面。
  • 支持模板背景部分变化。
  • 支持高分辨率。

快速启动

1. 本地安装: 环境检查/下载/安装

a. 环境检查

我们已验证EasyPhoto可在以下环境中执行：

Windows 10 的详细信息：

  • 操作系统： Windows10
  • python: python 3.10
  • pytorch: torch2.0.1
  • tensorflow-cpu: 2.13.0
  • CUDA: 11.7
  • CUDNN: 8+
  • GPU： Nvidia-3060 12G

Linux 的详细信息：

  • 操作系统 Ubuntu 20.04, CentOS
  • python: python3.10 & python3.11
  • pytorch: torch2.0.1
  • tensorflow-cpu: 2.13.0
  • CUDA: 11.7
  • CUDNN: 8+
  • GPU： Nvidia-A10 24G & Nvidia-V100 16G & Nvidia-A100 40G

我们需要大约 60GB 的可用磁盘空间（用于保存权重和数据集），请检查！

b. 代码下载

# 下载与安装
git clone https://github.com/aigc-apps/EasyPhoto.git
cd EasyPhoto
pip install -r requirements.txt

# 启动工具
python app.py

2. 从docker开始

# pull image
docker pull registry.cn-shanghai.aliyuncs.com/pai-ai-test/eas-service:easyphoto-diffusers-py310-torch201-cu117

# enter image
docker run -it --network host --gpus all registry.cn-shanghai.aliyuncs.com/pai-ai-test/eas-service:easyphoto-diffusers-py310-torch201-cu117

# launch
python app.py

如何使用

1. 模型训练

EasyPhoto训练界面如下：

  • 左边是训练图像。只需点击上传照片即可上传图片，点击清除照片即可删除上传的图片；
  • 右边是训练参数，不能为第一次训练进行调整。

点击上传照片后，我们可以开始上传图像这里最好上传5到20张图像，包括不同的角度和光照。最好有一些不包括眼镜的图像。如果所有图片都包含眼镜眼镜，则生成的结果可以容易地生成眼镜。 train_1

然后我们点击下面的“开始培训”，此时，我们需要填写上面的用户ID，例如用户名，才能开始培训。 train_2

模型开始训练后，webui会自动刷新训练日志。如果没有刷新，请单击“Refresh Log”按钮。 train_3

如果要设置参数，每个参数的解析如下：

参数名 含义
resolution 训练时喂入网络的图片大小，默认值为512
validation & save steps 验证图片与保存中间权重的steps数，默认值为100，代表每100步验证一次图片并保存权重
max train steps 最大训练步数，默认值为800
max steps per photos 每张图片的最大训练次数，默认为200
train batch size 训练的批次大小，默认值为1
gradient accumulationsteps 是否进行梯度累计，默认值为4，结合train batch size来看，每个Step相当于喂入四张图片
dataloader num workers 数据加载的works数量，windows下不生效，因为设置了会报错，Linux正常设置
learning rate 训练Lora的学习率，默认为1e-4
rank Lora 权重的特征长度，默认为128
network alpha Lora训练的正则化参数，一般为rank的二分之一，默认为64

2. 人物生成

a. 单人模版

  • 步骤1：点击刷新按钮，查询训练后的用户ID对应的模型。
  • 步骤2：选择用户ID。
  • 步骤3：选择需要生成的模板。
  • 步骤4：单击“生成”按钮生成结果。

single_people

b. 多人模板

  • 步骤1：转到EasyPhoto的设置页面，设置num_of_Faceid大于1。
  • 步骤2：应用设置。
  • 步骤3：重新启动webui的ui界面。
  • 步骤4：返回EasyPhoto并上传多人模板。
  • 步骤5：选择两个人的用户ID。
  • 步骤6：单击“生成”按钮。执行图像生成。

multi_people

算法详细信息

1. 架构概述

overview

在人工智能肖像领域，我们希望模型生成的图像逼真且与用户相似，而传统方法会引入不真实的光照（如人脸融合或roop）。为了解决这种不真实的问题，我们引入了稳定扩散模型的图像到图像功能。生成完美的个人肖像需要考虑所需的生成场景和用户的数字二重身。我们使用一个预先准备好的模板作为所需的生成场景，并使用一个在线训练的人脸 LoRA 模型作为用户的数字二重身，这是一种流行的稳定扩散微调模型。我们使用少量用户图像来训练用户的稳定数字二重身，并在推理过程中根据人脸 LoRA 模型和预期生成场景生成个人肖像图像。

2. 训练细节

overview

首先，我们对输入的用户图像进行人脸检测，确定人脸位置后，按照一定比例截取输入图像。然后，我们使用显著性检测模型和皮肤美化模型获得干净的人脸训练图像，该图像基本上只包含人脸。然后，我们为每张图像贴上一个固定标签。这里不需要使用标签器，而且效果很好。最后，我们对稳定扩散模型进行微调，得到用户的数字二重身。

在训练过程中，我们会利用模板图像进行实时验证，在训练结束后，我们会计算验证图像与用户图像之间的人脸 ID 差距，从而实现 Lora 融合，确保我们的 Lora 是用户的完美数字二重身。

此外，我们将选择验证中与用户最相似的图像作为 face_id 图像，用于推理。

3. 推理细节

a. 第一次扩散：

首先，我们将对接收到的模板图像进行人脸检测，以确定为实现稳定扩散而需要涂抹的遮罩。然后，我们将使用模板图像与最佳用户图像进行人脸融合。人脸融合完成后，我们将使用上述遮罩对融合后的人脸图像进行内绘（fusion_image）。此外，我们还将通过仿射变换（replace_image）把训练中获得的最佳 face_id 图像贴到模板图像上。然后，我们将对其应用 Controlnets，在融合图像中使用带有颜色的 canny 提取特征，在替换图像中使用 openpose 提取特征，以确保图像的相似性和稳定性。然后，我们将使用稳定扩散（Stable Diffusion）结合用户的数字分割进行生成。

b. 第二次扩散：

在得到第一次扩散的结果后，我们将把该结果与最佳用户图像进行人脸融合，然后再次使用稳定扩散与用户的数字二重身进行生成。第二次生成将使用更高的分辨率。

参考文献

相关项目

我们还列出了一些很棒的开源项目以及任何你可能会感兴趣的扩展项目：

许可证

本项目采用 Apache License (Version 2.0).

联系我们

  1. 使用钉钉搜索群38250008552或扫描下列二维码加入群聊
  2. 由于 微信群 已经满了，需要扫描右边的图片先添加这个同学为好友，然后再加入 微信群 。