教大家本地搭建一键换装AI开源工具 OOTDiffusion

前言

最近休息了两天,在研究一些AI用于商业化方向的东西,最近找到服装一键更换的开源项目,效果非常不错,今天就把安装步骤分享给大家,同时给大家看看真实效果

部署

项目地址:github.com/levihsu/OOT...

没有中文的项目介绍

首先需要拉取项目的所有代码:

shell 复制代码
git clone https://github.com/levihsu/OOTDiffusion

这个项目比较吃配置,机器配置是 内存 32G DDR4 显卡 2070S 8G CPU 9700F 系统 Ubuntu22.04

基础环境搭建,我这里的显卡驱动已经安装好了, 这里的环境是Ubuntu,可以参考我之前的本地部署ChatGLM2的文章安装显卡环境。

显卡相关的环境解决过后就需要使用conda 安装相关的依赖,现需要启动一个Python虚拟环境。

shell 复制代码
conda create -n ootd python==3.10
conda activate ootd
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
pip install -r requirements.txt

根据提示需要下载模型 ,模型下载地址有两个

shell 复制代码
git lfs clone  https://huggingface.co/levihsu/OOTDiffusion
git lfs clone  https://huggingface.co/openai/clip-vit-large-patch14

下载完成后移动到主项目的 checkpoints 目录,模型非常大,需要耗费很长的时间,下载完成后效果如下:

接下来需要启动web 服务,不推荐使用官方的命令行。启动web服务需要修改一下run/gradio_ootd.py,需要注销掉3行代码,默认 gradio 会调用两个显卡。

python 复制代码
# openpose_model_dc = OpenPose(1)
# parsing_model_dc = Parsing(1)
# ootd_model_dc = OOTDiffusionDC(1)

启动:

shell 复制代码
python run/gradio_ootd.py

本地访问:127.0.0.1:7865: 尝试一下

把淘宝的衣服下载下来尝试了一下,效果真不错哦

还可以调整,随机种子,重绘次数,一次生成的参数,但是目前仅仅生成上半身,上面注释的就是注释掉全身替换的代码,需要两块显卡同时运行。目前电脑只有一张显卡只能这样了

最后

我觉得这个效果非常炸裂,像这样的而且有一定的商业价值的项目我觉得非常有前景,对于一些想开服装网店的创业者非常有吸引力,可以节省大量的时间和金钱成本,对于这些开源进行的封装推广应该也有一定的商业价值,感兴趣的同学可以持续关注我,我会陆续介绍类似的项目。

今天就分享到这里,下个项目我在尝试声音克隆的本地部署的开源项目,敬请期待~

最后如果觉得内容不错有意思有用,欢迎点个关注,你的支持是我更新的动力~ 我自己本地部署了一个 可以直接体验(七天有效) :ootd.tuwei.space/

相关推荐
Easonmax2 小时前
用 Rust 打造可复现的 ASCII 艺术渲染器:从像素到字符的完整工程实践
开发语言·后端·rust
百锦再2 小时前
选择Rust的理由:从内存管理到抛弃抽象
android·java·开发语言·后端·python·rust·go
小羊失眠啦.2 小时前
深入解析Rust的所有权系统:告别空指针和数据竞争
开发语言·后端·rust
q***71853 小时前
Spring Boot 集成 MyBatis 全面讲解
spring boot·后端·mybatis
大象席地抽烟3 小时前
使用 Ollama 本地模型与 Spring AI Alibaba
后端
程序员小假3 小时前
SQL 语句左连接右连接内连接如何使用,区别是什么?
java·后端
小坏讲微服务3 小时前
Spring Cloud Alibaba Gateway 集成 Redis 限流的完整配置
数据库·redis·分布式·后端·spring cloud·架构·gateway
方圆想当图灵4 小时前
Nacos 源码深度畅游:Nacos 配置同步详解(下)
分布式·后端·github
方圆想当图灵4 小时前
Nacos 源码深度畅游:Nacos 配置同步详解(上)
分布式·后端·github
小羊失眠啦.4 小时前
用 Rust 实现高性能并发下载器:从原理到实战
开发语言·后端·rust