关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>

Yiwen2022 b4b17f33b9 更新 'requirements.txt'		11 months ago
configs	删除 'configs/wukong-huahua_inpaint_inference.yaml'	11 months ago

db_data	db_data commit	11 months ago

ldm	删除 'ldm/data/dataset.py'	11 months ago

show_imgs	上传文件至 'show_imgs'	11 months ago

README.md	更新 'README.md'	11 months ago

inference.py	更新 'inference.py'	11 months ago

requirements.txt	更新 'requirements.txt'	11 months ago

train.py	更新 'train.py'	11 months ago

蓉宝绘图

模型说明

Wukong-Huahua是基于扩散模型的中文文生图大模型，由华为诺亚团队携手中软分布式并行实验室，昇腾计算产品部联合开发。（https://github.com/mindspore-lab/minddiffusion/tree/main/vision/wukong-huahua）

启智平台：智算中心
公开模型搜索：rongbao
AI引擎：mindspore_2.0_train
第三方依赖：requirements.txt

本模型基于Wukong-Huahua，选择成都大运会吉祥物蓉宝的形象，能够基于3-5张同一主体的照片，经过25-35分钟的个性化微调，得到该主体定制化的图片生成模型。

效果展示,生成各种风格的主体图片：

下载Wukong-Huahua预训练参数 wukong-huahua-ms.ckpt 至 wukong-huahua/models/ 目录
训练数据，3-5张同一主体的照片（训练照片规格为512*512，尽量选取单一干净的背景，主体突出）
准备正则数据200张。如训练主体为狗，则需准备200张各种其他狗的图片，这些图片可以通过通用模型生成，也可以手动收集。提供了男人、女人、狗、猫四个类别的正则数据各200张。点击这里下载

本项目选择的是主体是：蓉宝

训练数据：
db_data/train

正则数据：
db_data/regular

训练作业

推理任务

No Description

Python Text