CatVTON是一个小型的AI虚拟换衣模型,适合每一个时尚爱好者使用。CatVTON 的特点是拥有轻量级网络,总共有899.06M 的参数,训练时只需要49.57M 的可训练参数。而且在进行推理时,使用的显存少于8G,支持1024x768的高分辨率,非常适合个人电脑进行操作。
产品入口:https://github.com/Zheng-Chong/CatVTON
产品特点可总结如下:
1) 轻量级网络(总共899.06M 参数)
2) 参数高效训练(49.57M 参数可训练)
3) 简化推理(<8G VRAM 分辨率为1024X768)
CatVTON 的开发团队最近在 GitHub 上发布了最新的代码和部署流程,包括如何在 ComfyUI 上快速部署 CatVTON。只需要简单的几步,就能让你在家中体验到最新的虚拟试妆技术。
安装步骤:
首先,你需要根据安装指南设置好环境,然后下载 ComfyUI-CatVTON 的文件,并将其解压到 ComfyUI 项目的 custom_nodes 文件夹中。完成这些步骤后,启动 ComfyUI,你就能享受到时尚搭配的乐趣了。
当然,如果你更喜欢使用 Gradio 应用,只需运行一条命令,系统会自动从 HuggingFace 下载所需的检查点,省时又省力。无论你是想在 DressCode 还是 VITON-HD 数据集上进行推理,CatVTON 都能轻松满足你的需求,相关的推理命令也非常简便。只需在命令行中输入相应的指令,几分钟后就能看到自己喜欢的效果。
此外,CatVTON 还支持多种精度选项,确保用户在不同的硬件条件下都能获得最佳体验。这个模型使用的是基于 Stable Diffusion v1.5的图像修复技术,结合了 SCHP 和 DensePose,能自动生成面具,帮助你更好地进行试妆。