返回模型

说明文档

IDM-VTON：改进扩散模型以实现真实场景下的虚拟试穿

这是论文《Improving Diffusion Models for Authentic Virtual Try-on in the Wild》的官方实现

🤗 试用我们的 HuggingFace 演示

teaser teaser2

待办事项

[x] 演示模型
[x] 推理代码
[ ] 训练代码

致谢

演示所用的 GPU 由 zerogpu 提供，自动掩码生成代码基于 OOTDiffusion 和 DCI-VTON。部分代码基于 IP-Adapter。

引用

@article{choi2024improving,
  title={Improving Diffusion Models for Virtual Try-on},
  author={Choi, Yisol and Kwak, Sangkyung and Lee, Kyungmin and Choi, Hyungwon and Shin, Jinwoo},
  journal={arXiv preprint arXiv:2403.05139},
  year={2024}
}

许可证

本仓库中的代码和模型权重采用 CC BY-NC-SA 4.0 许可证。

vanhieunguyen21/IDM-VTON

作者 vanhieunguyen21

image-to-image diffusers

↓ 1 ♥ 0

创建时间: 2025-03-21 07:22:12+00:00

更新时间: 2025-03-21 07:44:55+00:00

在 Hugging Face 上查看

文件 (30)

.gitattributes

README.md

assets/teaser.png

assets/teaser2.png

densepose/model_final_162be9.pkl

humanparsing/parsing_atr.onnx ONNX

humanparsing/parsing_lip.onnx ONNX

image_encoder/config.json

image_encoder/model.safetensors

model_index.json

openpose/ckpts/body_pose_model.pth

scheduler/scheduler_config.json

text_encoder/config.json

text_encoder/model.safetensors

text_encoder_2/config.json

text_encoder_2/model.safetensors

tokenizer/merges.txt

tokenizer/special_tokens_map.json

tokenizer/tokenizer_config.json

tokenizer/vocab.json

tokenizer_2/merges.txt

tokenizer_2/special_tokens_map.json

tokenizer_2/tokenizer_config.json

tokenizer_2/vocab.json

unet/config.json

unet/diffusion_pytorch_model.bin

unet_encoder/config.json

unet_encoder/diffusion_pytorch_model.safetensors

vae/config.json

vae/diffusion_pytorch_model.safetensors

说明文档

更多代码请查看我们的 GitHub 仓库！

IDM-VTON：改进扩散模型以实现真实场景下的虚拟试穿

待办事项

致谢

引用

许可证

vanhieunguyen21/IDM-VTON

文件 (30)