計畫簡介
OOTDiffusion專註於結合服裝融合和潛在擴散技術，為虛擬試穿提供可控制的解決方案。該計畫基於VITON-HD數據集訓練，實作了高解析度的虛擬試穿效果。計畫已經釋出了相關模型，並提供了完整的安裝和推理指南，幫助使用者輕松部署和使用該技術。
掃碼加入交流群
獲得更多技術支持和交流
此計畫可落地商用
流程
該計畫支持半身替換和全身替換，替換的效果我已經試用嘗試了，如果你也想試一下的話，可以在文章最後找到試用連結。
生成的效果可以說是秒殺之前市面上的所有軟體，已經達到了可以商用的水準。
範例
安裝
1.複制儲存庫
git clone https://github.com/levihsu/OOTDiffusion
2.建立一個 conda 環境並安裝所需的包
conda create -n ootd python==3.10conda activate ootdpip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 numpy==1.24.4 scipy==1.10.1 scikit-image==0.21.0 opencv-python==4.7.0.72 pillow==9.4.0 diffusers==0.24.0 transformers==4.36.2 accelerate==0.26.1 matplotlib==3.7.4 tqdm==4.64.1 gradio==4.16.0 config==0.5.1 einops==0.7.0 ninja==1.10.2
推理
1.半身模型
cd OOTDiffusion/runpython run_ootd.py --model_path <model-image-path> --cloth_path <cloth-image-path> --scale 2.0 --sample 4
2.全身模型
服裝類別必須配對：0 = 上半身；1 = 下半身；2 = 連衣裙
cd OOTDiffusion/runpython run_ootd.py --model_path <model-image-path> --cloth_path <cloth-image-path> --model_type dc --category 2 --scale 2.0 --sample 4
試用連結
https://ootd.ibot.cn/
關註「 開源AI計畫落地 」公眾號
