OOTDiffusion でバーチャル試着に挑戦

はじめに

バーチャル試着ができるという以下を使わせて頂きました。
github.com
非常にうまくいきます。
そのため、画像の捏造に利用されそうです。
悪用しないように気を付けて下さい。

結果




背景の変更には以前紹介した「Matte-Anything」を使っています。
touch-sp.hatenablog.com
元画像の女性は「fudukiMix_v20」で作成しています。

PC環境

Ubuntu 22.04 on WSL2
CUDA 11.8
Python 3.11

Windowsで動かそうとしましたが以下のエラーが出ました。

subprocess.CalledProcessError: Command '['where', 'cl']' returned non-zero exit status 1.

調べたところVisual Studioの環境変数の問題のようです。
Windows環境をあれこれ汚したくなかったのでWindowsでの実行はあきらめました。

Python環境構築

pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118
pip install 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

実行

GitHubに書いてある通りに実行しました。
Windowsで躓いた以外は問題なく進みました。

python run_ootd.py \
--model_path woman.png \
--cloth_path wear.jpg \
--scale 2.0 --sample 4





このエントリーをはてなブックマークに追加