StableVITON

https://gyazo.com/4176a4ed2748eb783e6ac7b396ab9eed

ComfyUI実装

weightは公式repoから申請する必要がある

なぜかcheckpointを借りられたのでComfyUIで試すnomadoor.icon

$ path\to\ComfyUI\custom_nodes\ComfyUI_Seg_VITON\checkpoints

VITONHD.ckptとsegformer_b2_clothesを置く

トップスやキャップ等、服の部位ごとにセグメンテーションができる

$ git lfs install

pipインストール

$ cd path\to\ComfyUI\custom_nodes\ComfyUI_Seg_VITON

$ ../../../python_embeded/python.exe -s -m pip install -r requirements.txt

実行するとエラーが出る

$ ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'

https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/11458を参考にpath\to\ComfyUI\custom_nodes\ComfyUI_Seg_VITON\ldm\models\diffusion\ddpm.pyの以下の部分を書き換える

code:ddpm.py

20 - from pytorch_lightning.utilities.distributed import rank_zero_only

20 + from pytorch_lightning.utilities.rank_zero import rank_zero_only

https://gyazo.com/7afcceff36c1cffc67ab2dba7b98e96b

Batch Sizeが最低16なせいかありえないくらい重い