返回模型

说明文档

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

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 许可证。

Roopansh/Ailusion-VTON-DEMO-v1.1

作者 Roopansh

image-to-image diffusers

↓ 0 ♥ 1

创建时间: 2024-05-16 07:14:31+00:00

更新时间: 2024-05-16 17:14:18+00:00

在 Hugging Face 上查看

文件 (145)

.gitattributes

.gitignore

README.md

assets/teaser.png

assets/teaser2.png

densepose/__init__.py

densepose/config.py

densepose/converters/__init__.py

densepose/converters/base.py

densepose/converters/builtin.py

densepose/converters/chart_output_hflip.py

densepose/converters/chart_output_to_chart_result.py

densepose/converters/hflip.py

densepose/converters/segm_to_mask.py

densepose/converters/to_chart_result.py

densepose/converters/to_mask.py

densepose/data/__init__.py

densepose/data/build.py

densepose/data/combined_loader.py

densepose/data/dataset_mapper.py

densepose/data/datasets/__init__.py

densepose/data/datasets/builtin.py

densepose/data/datasets/chimpnsee.py

densepose/data/datasets/coco.py

densepose/data/datasets/dataset_type.py

densepose/data/datasets/lvis.py

densepose/data/image_list_dataset.py

densepose/data/inference_based_loader.py

densepose/data/meshes/__init__.py

densepose/data/meshes/builtin.py

densepose/data/meshes/catalog.py

densepose/data/samplers/__init__.py

densepose/data/samplers/densepose_base.py

densepose/data/samplers/densepose_confidence_based.py

densepose/data/samplers/densepose_cse_base.py

densepose/data/samplers/densepose_cse_confidence_based.py

densepose/data/samplers/densepose_cse_uniform.py

densepose/data/samplers/densepose_uniform.py

densepose/data/samplers/mask_from_densepose.py

densepose/data/samplers/prediction_to_gt.py

densepose/data/transform/__init__.py

densepose/data/transform/image.py

densepose/data/utils.py

densepose/data/video/__init__.py

densepose/data/video/frame_selector.py

densepose/data/video/video_keyframe_dataset.py

densepose/engine/__init__.py

densepose/engine/trainer.py

densepose/evaluation/__init__.py

densepose/evaluation/d2_evaluator_adapter.py

densepose/evaluation/densepose_coco_evaluation.py

densepose/evaluation/evaluator.py

densepose/evaluation/mesh_alignment_evaluator.py

densepose/evaluation/tensor_storage.py

densepose/model_final_162be9.pkl

densepose/modeling/__init__.py

densepose/modeling/build.py

densepose/modeling/confidence.py

densepose/modeling/cse/__init__.py

densepose/modeling/cse/embedder.py

densepose/modeling/cse/utils.py

densepose/modeling/cse/vertex_direct_embedder.py

densepose/modeling/cse/vertex_feature_embedder.py

densepose/modeling/densepose_checkpoint.py

densepose/modeling/filter.py

densepose/modeling/hrfpn.py

densepose/modeling/hrnet.py

densepose/modeling/inference.py

densepose/modeling/losses/__init__.py

densepose/modeling/losses/chart.py

densepose/modeling/losses/chart_with_confidences.py

densepose/modeling/losses/cse.py

densepose/modeling/losses/cycle_pix2shape.py

densepose/modeling/losses/cycle_shape2shape.py

densepose/modeling/losses/embed.py

densepose/modeling/losses/embed_utils.py

densepose/modeling/losses/mask.py

densepose/modeling/losses/mask_or_segm.py

densepose/modeling/losses/registry.py

densepose/modeling/losses/segm.py

densepose/modeling/losses/soft_embed.py

densepose/modeling/losses/utils.py

densepose/modeling/predictors/__init__.py

densepose/modeling/predictors/chart.py

densepose/modeling/predictors/chart_confidence.py

densepose/modeling/predictors/chart_with_confidence.py

densepose/modeling/predictors/cse.py

densepose/modeling/predictors/cse_confidence.py

densepose/modeling/predictors/cse_with_confidence.py

densepose/modeling/predictors/registry.py

densepose/modeling/roi_heads/__init__.py

densepose/modeling/roi_heads/deeplab.py

densepose/modeling/roi_heads/registry.py

densepose/modeling/roi_heads/roi_head.py

densepose/modeling/roi_heads/v1convx.py

densepose/modeling/test_time_augmentation.py

densepose/modeling/utils.py

densepose/structures/__init__.py

densepose/structures/chart.py

densepose/structures/chart_confidence.py

densepose/structures/chart_result.py

densepose/structures/cse.py

densepose/structures/cse_confidence.py

densepose/structures/data_relative.py

densepose/structures/list.py

densepose/structures/mesh.py

densepose/structures/transform_data.py

densepose/utils/__init__.py

densepose/utils/dbhelper.py

densepose/utils/logger.py

densepose/utils/transform.py

densepose/vis/__init__.py

densepose/vis/base.py

densepose/vis/bounding_box.py

densepose/vis/densepose_data_points.py

densepose/vis/densepose_outputs_iuv.py

densepose/vis/densepose_outputs_vertex.py

densepose/vis/densepose_results.py

densepose/vis/densepose_results_textures.py

densepose/vis/extractor.py

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