OOTDiffusion 虚拟试穿部署完整教程：从环境搭建到出图

OOTDiffusion 虚拟试穿部署完整教程从环境搭建到出图【免费下载链接】OOTDiffusion[AAAI 2025] Official implementation of OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on项目地址: https://gitcode.com/GitHub_Trending/oo/OOTDiffusionOOTDiffusion 是基于潜在扩散模型的服装试穿项目能把一件衣服的图合成到一张模特图上支持半身上衣和全身上装、下装、连衣裙两种场景适合电商做商品展示或批量出图的开发者。功能速览输入一张模特图 一张衣服图输出一张穿上这件衣服的模特图。提供半身模型hd和全身模型dc两套权重由 run/run_ootd.py 一键推理。自带命令行脚本和 Gradio 网页界面无需写代码即可调参出图。前置依赖人体解析和姿态估计OpenPose仓库已内置推理模块。仓库在 Ubuntu 22.04 / Linux 上验证建议同环境部署。前置检查硬件需要一块带 CUDA 的 NVIDIA 显卡显存建议 8GB 起步--sample越大越吃显存。软件LinuxUbuntu 22.04 最稳已装 condaPython 需固定为 3.10。磁盘模型权重与依赖约需数 GB建议预留 10GB 以上。网络能访问 pip 源和模型权重下载源。部署三步走第一步取代码git clone https://gitcode.com/GitHub_Trending/oo/OOTDiffusion cd OOTDiffusion怎么确认成功进入目录后能看到run/、ootd/、preprocess/、checkpoints/四个目录其中 checkpoints/ 下只有一篇 README.txt说明还没放权重。第二步备环境conda create -n ootd python3.10 conda activate ootd pip install torch2.0.1 torchvision0.15.2 torchaudio2.0.2 pip install -r requirements.txt怎么确认成功激活环境后运行下面命令打印2.0.1 True第二项为 True 表示检测到 GPUpython -c import torch; print(torch.__version__, torch.cuda.is_available())依赖清单见 requirements.txt核心是diffusers、transformers、onnxruntime、gradio。第三步拿权重推理代码会到 checkpoints/ 下读取四组文件全部下齐并放进对应子目录权重目录位置作用ootdcheckpoints/ootd/试穿主模型含ootd_hd/checkpoint-36000、ootd_dc/checkpoint-36000和 VAEhumanparsingcheckpoints/humanparsing/人体解析parsing_atr.onnx、parsing_lip.onnxopenposecheckpoints/openpose/ckpts/姿态估计body_pose_model.pthclip-vit-large-patch14checkpoints/clip-vit-large-patch14/文本/图像特征编码这些路径由 ootd/inference_ootd_hd.py、preprocess/humanparsing/run_parsing.py、preprocess/openpose/annotator/util.py 指定请严格按目录摆放。怎么确认成功ls checkpoints能看到ootd、humanparsing、openpose、clip-vit-large-patch14四组且各自的.onnx/.pth/ 权重文件都在位。首次运行先跑最小命令半身模型上衣cd run python run_ootd.py --model_path examples/model/model_1.png --cloth_path examples/garment/03244_00.jpg --scale 2.0 --sample 4怎么判断结果正常当前run/目录下会生成images_output/里面有mask.jpg和out_hd_0.png到out_hd_3.png。打开out_hd_*.png模特身上的衣服已换成指定上衣即为成功。全身模型连衣裙示例cd run python run_ootd.py --model_path examples/model/model_8.png --cloth_path examples/garment/048554_1.jpg --model_type dc --category 2 --scale 2.0 --sample 4若想要网页界面启动 Gradio见 run/gradio_ootd.pycd run python gradio_ootd.py浏览器访问http://localhost:7865可分别用半身、全身两块上传示例图调参。参数速查以下参数由 run/run_ootd.py 解析参数作用推荐值--model_type模型类型hd半身 /dc全身hd默认--category服装类别 0 上衣 / 1 下装 / 2 连衣裙须与model_type匹配0hd仅支持 0--scale引导强度越大越贴合衣服纹理2.0范围 1.0–5.0--step扩散步数越大越细腻越慢20范围 20–40--sample一次生成几张1范围 1–4--seed随机种子固定可复现结果-1随机常见问题自查现象运行报CUDA out of memory。原因显存不足--sample或--step偏大。解法把--sample降到 1--step降到 20。现象半身模式加--category 1或2时直接报错退出。原因hd模型只支持上衣代码对类别做了强校验。解法半身用--category 0要下装或连衣裙就换--model_type dc。现象启动报找不到parsing_atr.onnx、body_pose_model.pth或checkpoint-36000。原因checkpoints/ 下权重没下全或目录层级不对。解法按第三步拿权重的表格补全四组文件注意openpose还要进ckpts/子目录。现象能出图但衣服颜色/图案没换上去像是只换了个轮廓。原因引导强度偏低生成偏向原图。解法把--scale提到 2.5–3.0 再试。收尾到这里你已完成 OOTDiffusion 的代码拉取、环境搭建、权重放置与首次出图并掌握了命令行与 Gradio 两种用法。后续可阅读 ootd/pipelines_ootd/pipeline_ootd.py 了解推理管线或对照 run/utils_ootd.py 理解 mask 生成逻辑。【免费下载链接】OOTDiffusion[AAAI 2025] Official implementation of OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on项目地址: https://gitcode.com/GitHub_Trending/oo/OOTDiffusion创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考