凡心所向，素履所往，生如逆旅，一苇以航。——尘曲
https://github.com/yisol/IDM-VTON
分享一个AI换装虚拟试穿的模型
1
2
3
4
5
| git clone https://github.com/yisol/IDM-VTON.git
cd IDM-VTON
conda env create -f environment.yaml
conda activate idm
|
VITON-HD 数据集可以从这里下载
GitHub - shadow2496/VITON-HD: Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)
下完了把 vitonhd_test_tagged.json 放到 test 目录中
1
2
3
4
5
6
7
8
9
10
|
train
|-- ...
test
|-- image
|-- image-densepose
|-- agnostic-mask
|-- cloth
|-- vitonhd_test_tagged.json
|
推理
1
2
3
4
5
6
7
8
| accelerate launch inference.py \
--width 768 --height 1024 --num_inference_steps 30 \
--output_dir "result" \
--unpaired \
--data_dir "DATA_DIR" \
--seed 42 \
--test_batch_size 2 \
--guidance_scale 2.0
|
直接用脚本也行