教大家本地搭建一键换装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/

相关推荐
小江的记录本3 分钟前
【Java基础】Java 8-21新特性 :JDK17:密封类、模式匹配、Record类(附《思维导图》+《面试高频考点清单》)
java·数据结构·后端·python·mysql·面试·职场和发展
Mahir0810 分钟前
Spring 核心原理:IoC/DI 与 Bean 生命周期全景解析
java·后端·spring·面试·bean生命周期·控制反转ioc·依赖注入di
weixin_4896900211 分钟前
NAS部署实测:Solon vs Spring Boot,从内存到包体积的“降维打击”
java·spring boot·后端
枕星而眠17 分钟前
数据结构哈希表(散列表)超详细总结
c语言·数据结构·后端·散列表
一条泥憨鱼17 分钟前
【Java 进阶】LinkedHashMap 与 TreeMap
java·开发语言·数据结构·笔记·后端·学习
Lee川9 小时前
mini-cursor 揭秘:从 Tool 定义到 Agent 循环的完整实现
前端·人工智能·后端
星浩AI11 小时前
OpenHuman 对比 OpenClaw、Hermes Agent
人工智能·后端·agent
小江的记录本11 小时前
【Java基础】泛型:泛型擦除、通配符、上下界限定(附《思维导图》+《面试高频考点清单》)
java·数据结构·后端·mysql·spring·面试·职场和发展
geovindu13 小时前
go: Semaphore Pattern
开发语言·后端·设计模式·golang·企业级信号量模式
IT_陈寒13 小时前
Redis内存用爆了,原来我们都忽略了这个配置
前端·人工智能·后端