返回模型

说明文档

OOTDiffusion

我们的 OOTDiffusion GitHub 仓库

🤗 体验 OOTDiffusion

（感谢 ZeroGPU 提供 A100 GPU）

OOTDiffusion：基于服装融合的潜在扩散模型用于可控虚拟试穿 [arXiv 论文]<br> Yuhao Xu, Tao Gu, Weifeng Chen, Chengcai Chen<br> 小i研究院

我们在 VITON-HD（半身）和 Dress Code（全身）数据集上训练的模型权重已发布

📢📢 我们现在支持人体解析的 ONNX 版本。大多数环境问题应该已经解决 : )
请同时下载 clip-vit-large-patch14 到 checkpoints 文件夹
我们仅在 Linux（Ubuntu 22.04）上测试了代码和模型

工作流程

引用

@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}
}

levihsu/OOTDiffusion

作者 levihsu

diffusers

↓ 0 ♥ 329

创建时间: 2024-02-21 02:50:36+00:00

更新时间: 2024-04-17 06:03:25+00:00

在 Hugging Face 上查看

文件 (28)

.gitattributes

README.md

checkpoints/humanparsing/exp-schp-201908261155-lip.pth

checkpoints/humanparsing/exp-schp-201908301523-atr.pth

checkpoints/humanparsing/parsing_atr.onnx ONNX

checkpoints/humanparsing/parsing_lip.onnx ONNX

checkpoints/ootd/feature_extractor/preprocessor_config.json

checkpoints/ootd/model_index.json

checkpoints/ootd/ootd_dc/checkpoint-36000/unet_garm/config.json

checkpoints/ootd/ootd_dc/checkpoint-36000/unet_garm/diffusion_pytorch_model.safetensors

checkpoints/ootd/ootd_dc/checkpoint-36000/unet_vton/config.json

checkpoints/ootd/ootd_dc/checkpoint-36000/unet_vton/diffusion_pytorch_model.safetensors

checkpoints/ootd/ootd_hd/checkpoint-36000/unet_garm/config.json

checkpoints/ootd/ootd_hd/checkpoint-36000/unet_garm/diffusion_pytorch_model.safetensors

checkpoints/ootd/ootd_hd/checkpoint-36000/unet_vton/config.json

checkpoints/ootd/ootd_hd/checkpoint-36000/unet_vton/diffusion_pytorch_model.safetensors

checkpoints/ootd/scheduler/scheduler_config.json

checkpoints/ootd/text_encoder/config.json

checkpoints/ootd/text_encoder/pytorch_model.bin

checkpoints/ootd/tokenizer/merges.txt

checkpoints/ootd/tokenizer/special_tokens_map.json

checkpoints/ootd/tokenizer/tokenizer_config.json

checkpoints/ootd/tokenizer/vocab.json

checkpoints/ootd/vae/config.json

checkpoints/ootd/vae/diffusion_pytorch_model.bin

checkpoints/openpose/ckpts/body_pose_model.pth

images/demo.png

images/workflow.png