Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

CatVTON: Concatenation Is All You Need for Virtual Try-On with Diffusion Models

About

Virtual try-on methods based on diffusion models achieve realistic effects but often require additional encoding modules, a large number of training parameters, and complex preprocessing, which increases the burden on training and inference. In this work, we re-evaluate the necessity of additional modules and analyze how to improve training efficiency and reduce redundant steps in the inference process. Based on these insights, we propose CatVTON, a simple and efficient virtual try-on diffusion model that transfers in-shop or worn garments of arbitrary categories to target individuals by concatenating them along spatial dimensions as inputs of the diffusion model. The efficiency of CatVTON is reflected in three aspects: (1) Lightweight network. CatVTON consists only of a VAE and a simplified denoising UNet, removing redundant image and text encoders as well as cross-attentions, and includes just 899.06M parameters. (2) Parameter-efficient training. Through experimental analysis, we identify self-attention modules as crucial for adapting pre-trained diffusion models to the virtual try-on task, enabling high-quality results with only 49.57M training parameters. (3) Simplified inference. CatVTON eliminates unnecessary preprocessing, such as pose estimation, human parsing, and captioning, requiring only a person image and garment reference to guide the virtual try-on process, reducing over 49% memory usage compared to other diffusion-based methods. Extensive experiments demonstrate that CatVTON achieves superior qualitative and quantitative results compared to baseline methods and demonstrates strong generalization performance in in-the-wild scenarios, despite being trained solely on public datasets with 73K samples.

Zheng Chong, Xiao Dong, Haoxiang Li, Shiyue Zhang, Wenqing Zhang, Xujie Zhang, Hanqing Zhao, Dongmei Jiang, Xiaodan Liang• 2024

Related benchmarks

TaskDatasetResultRank
Virtual Try-OnVITON-HD paired
LPIPS0.057
58
Virtual Try-OnVITON-HD (test)
SSIM86.94
57
Virtual Try-OnVITON-HD unpaired
FID9.015
50
Image Virtual Try-onVITON-HD
LPIPS0.057
41
Virtual Try-OnDressCode (test)
FID3.2755
29
Virtual Try-OnVITON-HD Unpaired (test)
FID9.076
26
Virtual Try-OnDressCode Dresses (unpaired)
FID5.24
25
Virtual Try-OnVITON-HD 1024x768 (test)
SSIM87
21
Virtual Try-OnVITON paired HD (test)
FID5.42
19
Virtual Try-OnDressCode
LPIPS0.037
19
Showing 10 of 44 rows

Other info

Follow for update