AI换装虚拟试穿模型
凡心所向，素履所往，生如逆旅，一苇以航。——尘曲
https://github.com/yisol/IDM-VTON
分享一个AI换装虚拟试穿的模型
git clone https://github.com/yisol/IDM-VTON.git
cd IDM-VTON
conda env create -f environment.yaml
conda activate idmVITON-HD 数据集可以从这里下载
下完了把 vitonhd_test_tagged.json 放到 test 目录中
train
|-- ...
test
|-- image
|-- image-densepose
|-- agnostic-mask
|-- cloth
|-- vitonhd_test_tagged.json推理
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直接用脚本也行
sh inference.sh本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2024-05-25，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
