该项目在CPU环境下运行需要较大的内存（大概40G）

安装

github地址：https://github.com/levihsu/OOTDiffusion

huggingface地址：https://huggingface.co/spaces/levihsu/OOTDiffusion

获取源码

使用git拉取

set http_proxy=127.0.0.1:7890
set GIT_LFS_SKIP_SMUDGE=1
git clone https://huggingface.co/spaces/levihsu/OOTDiffusion

拉取huggingface需要使用代理

GIT_LFS_SKIP_SMUDGE=1会自动忽略大文件

获取模型权重

Python安装包

pip install huggingface_hub=0.25.2

使用huggingface-cli下载huggingface中的模型

下载OOTDiffusion模型

huggingface-cli download --resume-download levihsu/OOTDiffusion --local-dir ./model1 --local-dir-use-symlinks False

模型会下载到./models文件夹中，如果文件夹不存在会自动创建

下载clip-vit-large-patch14模型

huggingface-cli download --resume-download openai/clip-vit-large-patch14 --local-dir ./model2 --local-dir-use-symlinks False

下载完成后，将model1中的checkpoints复制到源码文件夹的根目录下

将model2重命名为clip-vit-large-patch14，并且放到源码文件夹下的checkpoints中

修改源码

模型类型分为hd和dc，因此需要修改对应的类OOTDiffusionHD、OOTDiffusionDC

if model_type == "hd":
    model = OOTDiffusionHD(args.gpu_id)
elif model_type == "dc":
    model = OOTDiffusionDC(args.gpu_id)

假设使用的是hd模式，需要修改OOTDiffusionHD

...
VIT_PATH = "./checkpoints/clip-vit-large-patch14"  # 修改为checkpoints中的clip-vit-large-patch14路径
...

运行

执行命令

python ./run/run_ootd.py --model_path 1.jpg --cloth_path 2.jpg --step 10 --scale 1.0 --sample 1 --gpu_id -1

–model_path：模特图片地址
–cloth_path：衣服地址
–step：迭代步长
–gpu_id：GPUID