地址:https://github.com/muzishen/IMAGDressing
环境:单卡运行即可
registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.1.0-py310-torch2.1.2-tf2.14.0-1.13.1
运行
下载模型放到ckpt
随机面孔和姿势来穿上指定的衣服:
python inference_IMAGdressing.py --cloth_path cloth/00008_00.jpg
可以修改随机数种子,生成不同的图片
随机面孔使用特定姿势搭配特定服装:
python inference_IMAGdressing_ipa_controlnetpose.py --cloth_path cloth/00008_00.jpg --pose_path image-densepose/00135_00.jpg
可以修改随机数种子,生成不同的图片