头像车车 创建于 2024年11月16日 修改于 2024年11月16日31次阅读

开源有趣的虚拟试穿再也不用为女朋友问那件衣服好看而发愁了

如何应对女朋友挑选衣服时的“送命题”

很多小伙伴肯定都遇到过女朋友或者老婆在挑选衣服时问哪一件衣服好看的问题。这是一个送命题,回答都好看自己的钱包不保,回答某一件或者都不好看女朋友的好心情不保,愁啊!

衣服选择

当然了,作为一个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
    最后一步取决于网络下载情况,需要几分钟至20多分钟不等。
安装过程

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

点赞
分享
打赏
评论
全部评论
加载失败