本地整合包免费下载

夸克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 安装

  1. 克隆地址
git clone https://github.com/levihsu/OOTDiffusion
  1. 创建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 推理

  1. 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
  1. 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

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。