开源有趣的虚拟试穿再也不用为女朋友问那件衣服好看而发愁了
如何应对女朋友挑选衣服时的“送命题”
很多小伙伴肯定都遇到过女朋友或者老婆在挑选衣服时问哪一件衣服好看的问题。这是一个送命题,回答都好看自己的钱包不保,回答某一件或者都不好看女朋友的好心情不保,愁啊!

当然了,作为一个IT男怎么能被这种问题难倒呢?今天偶然在Github上发现一个有趣的项目【OOTDiffusion】。
在线体验
此项目能够将图片和衣服进行搭配,虚拟换装,直接展示衣服试穿效果。官方提供了在线体验环境,小伙伴可以直接体验:
或者可直接看下方Demo视频。
本地部署
当然了,如果担心隐私问题,此项目可以部署在有显卡的服务器上。以下是部署步骤:
01. 部署环境说明
为了方便,本文中采用了腾讯云服务器,配置了一颗NVIDIA T4显卡,具体型号如下:

02. 开始安装部署
- 克隆仓库到本地服务器
bash
git clone https://github.com/levihsu/OOTDiffusion
- 安装环境和必要的软件包
bash
conda create -n ootd python==3.10 conda activate ootd pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 numpy==1.24.4 scipy==1.10.1 scikit-image==0.21.0 opencv-python==4.7.0.72 pillow==9.4.0 diffusers==0.24.0 transformers==4.36.2 accelerate==0.26.1 matplotlib==3.7.4 tqdm==4.64.1 gradio==4.16.0 config==0.5.1 einops==0.7.0 ninja==1.10.2

03. 尝试推理
半身推理模型
bash
cd OOTDiffusion/run
python run_ootd.py --model_path <model-image-path> --cloth_path <cloth-image-path> --scale 2.0 --sample 4
效果如下图:

全身推理模型
bash
cd OOTDiffusion/run
python run_ootd.py --model_path <model-image-path> --cloth_path <cloth-image-path> --model_type dc --category 2 --scale 2.0 --sample 4
效果如下图:

感兴趣的小伙伴实践起来吧,再也不用为女朋友或者老婆在挑选衣服的时候问哪一件衣服好看而发愁了。
end
全部评论
加载失败
暂无匹配数据