项目地址:https://github.com/levihsu/OOTDiffusion
试用地址:https://ootd.ibot.cn/
本地部署
下载模型
- git lfs安装, 然后国内源下载 git clone https://www.modelscope.cn/AI-ModelScope/clip-vit-large-patch14.git
- 然后国内镜像手动下载 https://hf-mirror.com/levihsu/OOTDiffusion/tree/main 相应模型文件(git 好像访问不到)
- 将模型文件放到项目checkpints下
代码修改
由于只有一个GPU,文件run\gradio_ootd.py中#24~#26原来为:
openpose_model_dc = OpenPose(1)
parsing_model_dc = Parsing(1)
ootd_model_dc = OOTDiffusionDC(1)
把它们改成:
openpose_model_dc = openpose_model_hd
parsing_model_dc = parsing_model_hd
ootd_model_dc = ootd_model_hd
代码运行
- 指定图片运行 python3 run_ootd.py --model_path ren.png --cloth_path clothes2.jpg --scale 2.0 --sample 4
- 网页demo运行 python3 gradio_ootd.py