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

最新推荐文章于 2024-08-19 16:40:05 发布

AI-椰子不椰

最新推荐文章于 2024-08-19 16:40:05 发布

阅读量748

点赞数 21

文章标签：人工智能语言模型 AI大模型大模型

本文链接：https://blog.csdn.net/2401_85343303/article/details/140605095

版权

前言

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

部署

项目地址：[github.com/levihsu/OOT…]

没有中文的项目介绍

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

git clone https://github.com/levihsu/OOTDiffusion

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

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

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

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

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

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 会调用两个显卡。

# openpose_model_dc = OpenPose(1)
# parsing_model_dc = Parsing(1)
# ootd_model_dc = OOTDiffusionDC(1)

启动：

python run/gradio_ootd.py

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

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

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

最后

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

如何学习AI大模型？

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

在这里插入图片描述