CatVTON免安装win-webui版
更新: 12/21/2024 字数: 0 字 时长: 0 分钟
概述
CatVTON是一个小型的AI虚拟换衣模型,适合每一个时尚爱好者使用。CatVTON 的特点是拥有轻量级网络,总共有899.06M 的参数,训练时只需要49.57M 的可训练参数。而且在进行推理时,使用的显存少于8G,支持1024x768的高分辨率,非常适合个人电脑进行操作。
特点
CatVTON是一个简单高效的虚拟试穿扩散模型,具有:
1)轻量级网络(完全899.06M参数)
2)参数高效训练(49.57M参数可训练)
3)简化推理(小于8G VRAM,分辨率为1024X768)
项目简介
项目信息
在线体验
HuggingFace体验:点击访问
模型下载
paper
arxiv: 点击访问
手动安装
安装
- 克隆仓库
bash
git clone https://github.com/Zheng-Chong/CatVTON
- 创建环境
bash
conda create -n catvton python==3.9.0
conda activate catvton
cd CatVTON-main # or your path to CatVTON project dir
pip install -r requirements.txt
- 实测可安装的requirements.txt
- torch使用cuda-2.1.2+121版本(实测2.0版本不可用)
- torch安装太慢,可考虑使用
https://mirror.sjtu.edu.cn/pytorch-wheels/torch_stable.html
torch==2.1.2+cu121
torchvision==0.16.2+cu121
-f https://download.pytorch.org/whl/torch_stable.html
#-f https://mirror.sjtu.edu.cn/pytorch-wheels/torch_stable.html
accelerate==0.31.0
diffusers==0.29.2
huggingface_hub==0.23.4
matplotlib
numpy==1.26.4
opencv_python==4.10.0.84
pillow==10.3.0
PyYAML==6.0.1
scipy==1.13.1
setuptools==51.0.0
scikit-image==0.24.0
tqdm==4.66.4
transformers==4.27.3
fvcore==0.1.5.post20221221
cloudpickle==3.0.0
omegaconf==2.3.0
pycocotools==2.0.8
av==12.3.0
gradio==4.41.0
# 解决默认安装的fastapi版本报错
fastapi==0.109.2
🔑万能君exe免安装环境版
注意
解压后,路径不要含有中文,路径不要含有中文,路径不要含有中文
说明
- 软件已经过测试,测试平台为
Windows10
和Nvidia-4090
显卡 - 不支持
AMD显卡
及核显
,显存尽量8GB
以上,cuda-12
版本,低显存
或低cuda版本不保证正常使用 - 点此查看自己的显卡相关信息
- 压缩包已包含依赖的环境模型等大文件,无需安装环境,点开即用;
- 体积较小,适合快速试用
- 大小:6GB
下载地址
主地址
备用地址
- 密码:4331
💎免安装环境win-webui版
注意
解压后,路径不要含有中文,路径不要含有中文,路径不要含有中文
说明
- 软件已经过测试,测试平台为
Windows10
和Nvidia-4090
显卡 - 不支持
AMD显卡
及核显
,显存尽量8GB
以上,cuda-12
版本,低显存
或低cuda版本不保证正常使用 - 点此查看自己的显卡相关信息
- 压缩包已包含依赖的环境模型等大文件,无需安装环境,点开即用;
- webUI版使用官方Git源码构建,功能更全面,可直接使用官方运行示例
- 大小:22GB
下载地址
主地址
备用地址
- 密码:4331
常见问题
本地访问报错
开启科学上网后,可能本地
localhost
和127.0.0.1
可能会访问出错- ValueError: When localhost is not accessible a shareable link must be created
设置环境变量
- sh
# linux设置 export no_proxy="localhost,127.0.0.1,::1" # windows“编辑环境变量”或在“运行.bat”文件中添加如下配置 set NO_PROXY="localhost,127.0.0.1"
实测
注意事项
- 图片背景尽量简单,
宽高
比例最好为0.75 - 上传的图片,例如:
裙子
,替换试穿的服装应该也为裙子
,否则AI可能将脑补出其他部分
ComfyUI工作流程
我们修改了主代码,以便在ComfyUI上轻松部署CatVTON。由于代码结构不兼容,我们在版本中发布了这部分,其中包括ComfyUI的custom_nodes
下的代码和我们的工作流JSON文件
要将CatVTON部署到您的ComfyUI,请执行以下步骤:
- 安装CatVTON和ComfyUI的所有要求,请参阅CatVTON安装指南和ComfyUI安装指南。
- 下载ComfyUI-CatVTON.zip并将其解压缩到ComfyUI项目下的custom_nodes文件夹中(从ComfyUI克隆)。
- 运行ComfyUI。
- 下载catvton_workflow. json并将其拖到您的ComfyUI网页并享受 😆