AI虚拟换衣模型CatVTON:参数小仅899.06M 支持高分辨率
CatVTON是一款专为时尚爱好者设计的轻量级AI虚拟换衣模型,总参数为899.06M,训练时只需49.57M可训练参数。推理时显存需求低于8G,支持1024x768高分辨率,适合个人电脑操作。该模型具备在ComfyUI和Gradio应用上快速部署的能力,支持VITON-HD和DressCode数据集的推理,并提供多种精度选项,以适应不同硬件条件。CatVTON利用基于Stable Diffusion v1.5的图像修复技术,结合SCHP和DensePose,能自动生成面具,提升试妆体验。