本地整合包免费下载
夸克:https://pan.quark.cn/s/79b80dfcb424
迅雷:https://pan.xunlei.com/s/VNxbwX-IbsEB64HcwVeHOVPMA1?pwd=6ws3#
简介
今天给大家分享一个OOTDiffusion小工具,只需上传模特图片和衣服图片,就能实现一键自动换装,再也不用像webui和comfui还要去调整很多参数,
OOTDiffusion 正式实施:基于舾装融合的潜在扩散用于可控虚拟试穿
该工具基于Github作者:OOTDiffusion 开源项目整合而来,工具目前有三个功能:更换上半身、下半身和全身换装。
Half-body 半身:这里也给我们提供了一些参考样图,左边选择模特,中间衣服,images设置每次跑多少张,Steps步数,指导尺度根据需求来,最后再点击Run生成
Full-body全身:该模式下提供了三种服装类别,Upper-body、Lower-body、Dress,
根据自己的需求来选择,比如你要换上衣,那么你就选择Upper-body(上半身),裤子就选择Lower-body,裙子(Dress)
注意:当然你还可以之定义上传自己的产品图片,但模特的尺寸一定要按照作者提供的尺寸来3:4竖图,不然换出来要错位,
衣服和模特图片建议尺寸(768x1024),底色也按照推荐样式来。
在线体验地址:
https://huggingface.co/spaces/levihsu/OOTDiffusion(基于A100GPU,最好科学上网)
原理
Installation 安装
- 克隆地址
git clone https://github.com/levihsu/OOTDiffusion
- 创建conda环境并安装所需的包
conda create -n ootd python==3.10 conda activate ootd pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pip install -r requirements.txt
Inference 推理
- Half-body model 半身模型
cd OOTDiffusion/run python run_ootd.py --model_path <model-image-path> --cloth_path <cloth-image-path> --scale 2.0 --sample 4
- Full-body model 全身模型
服装类别必须配对:0 = 上半身; 1 = 下半身; 2 = 连衣裙
cd OOTDiffusion/run python run_ootd.py --model_path <model-image-path> --cloth_path <cloth-image-path> --model_type dc --category 2 --scale 2.0 --sample 4
Citation 引文
@article{xu2024ootdiffusion, title={OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on}, author={Xu, Yuhao and Gu, Tao and Chen, Weifeng and Chen, Chengcai}, journal={arXiv preprint arXiv:2403.01779}, year={2024} }
本地整合包安装
该整合包由B站作者:万能的软件库整合而来,解压即用,不用部署其他环境。
下载解压后里面有2个版本,v0.1和v1.0,v0.1版本需要较高的显存,webui操作界面,如果你要用webui界面,请运行“AI自动换装OOTDiffusion_V0.1(WebUI界面).exe”,
选择你需要的模式,然后启动webui(注意:终端不能关闭)
而v1.0版本增加了低显存模型,据整合包作者介绍4G显存即可运行,自定义界面请运行AI自动换装OOTDiffusion_V1.0(软件界面).exe
选择人物图片、衣服图片,开启低显存模式,调整其他参数,最后点击开始换装(注意:终端不能关闭)
整合包更新记录
V1.0版本:2024-2-27
- 去掉webui界面方式,改为界面方式;
- 支持批量选择图片,批量转换;
- 低显存占用模式,最新显卡驱动加持下,4G显存就能跑;
- 支持16代显卡等不支持半精度fp16显卡运行。
V0.1版本:2024-2-25
- 尝鲜版本,为webui启动方式,后续做成纯界面调用方式。
如遇失效,请加v:xkd2310备注SD