阿里云人工智能平台PAIAI个人写真-云淘科技

本文为您介绍如何在交互式建模DSW开发环境中，利用ModelScope社区的个人写真开源版本FaceChain，实现AI个人写真。

前提条件

已开通DSW并创建默认工作空间，详情请参见开通并创建默认工作空间。

步骤一：准备数据

请准备3-10张头肩照，确保图片中的面部清晰可见，避免多人脸或脸部遮挡等情况，以免影响效果。

步骤二：创建并进入DSW实例页面

创建DSW实例，其中关键参数配置如下，更多参数配置说明，请参见创建及管理DSW实例。
- 资源组：DSW实例的显存要求不小于20 GB。本案例选择公共资源组-按量付费>GPU规格>ecs.gn7i-c8g1.2xlarge。
- 选择镜像：选择官方镜像>modelscope:ubuntu20.04-cuda11.7.1-py38-torch2.0.1-tf1.15.5-1.8.1。
单击目标实例操作列下的打开。

在顶部菜单栏，单击Terminal页签，使用以下命令查看显存是否满足要求。

nvidia-smi

回显结果如下。显存为22731 MiB，符合要求。

Fri Sep  1 16:11:43 2023       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.82.01    Driver Version: 470.82.01    CUDA Version: 11.7     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA A10          Off  | 00000000:00:08.0 Off |                    0 |
|  0%   29C    P8     9W / 150W |      0MiB / 22731MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

步骤三：搭建体验环境

在Terminal中执行以下命令，克隆Git代码并安装依赖库。

git clone https://www.modelscope.cn/studios/CVstudio/cv_human_portrait.git
cd cv_human_portrait
pip install -r requirements.txt
pip install gradio==3.35.2

在Terminal中执行以下命令，来启动应用程序。

python app.py

回显结果如下。

023-09-01 16:15:42,668 - modelscope - INFO - PyTorch version 2.0.1+cu117 Found.
2023-09-01 16:15:42,673 - modelscope - INFO - Loading ast index from /mnt/workspace/.cache/modelscope/ast_indexer
2023-09-01 16:15:42,673 - modelscope - INFO - No valid ast index found from /mnt/workspace/.cache/modelscope/ast_indexer, generating ast index from prebuilt!
2023-09-01 16:15:42,924 - modelscope - INFO - Loading done! Current index file version is 1.8.1, with md5 2c26b8f616d95476e0bf242e6c0cabb6 and a total number of 893 components indexed
[2023-09-01 16:15:44,791] [INFO] [real_accelerator.py:133:get_accelerator] Setting ds_accelerator to cuda (auto detect)
2023-09-01 16:15:45.808875: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
WARNING:tensorflow:Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them.
/opt/conda/lib/python3.8/site-packages/gradio/components/gallery.py:205: UserWarning: The `style` method is deprecated. Please set these arguments in the constructor instead.
  warnings.warn(
Running on local URL:  http://127.0.0.1:7860

To create a public link, set `share=True` in `launch()`.

在步骤2的回显结果中，单击local URL后的http://127.0.0.1:7860，启动WebUI页面。

说明

由于http://127.0.0.1:7860为内网访问地址，仅支持在当前的DSW实例内部通过单击链接来访问WebUI页面，不支持通过外部浏览器直接访问。

步骤四：推理模型生成写真

在形象定制（Train）页签的训练图片（Training photos）区域中，单击选择图片上传（Upload photos），并根据界面提示上传已准备好的图片。
单击开始训练。

大约需要持续5分钟，即可训练完成。
当提示训练已经完成时，在WebUI页面上方，切换到形象体验（Inference）页签。在该页面根据需要配置风格模型、服装风格等参数，然后单击开始生成（Start!）。

大约需要等待5分钟，即可在生成结果（Result）区域查看到推理结果。以工作服（working suit）为例，效果图如下。