MaX4Zero: Masked Extended Attention for
Zero-Shot Virtual Try-On In The Wild


Tel Aviv University
Preview of MaX4Zero Virtual Try-On Results

MaX4Zero performs Virtual Try-On in the wild (for unseen target images and garments) without any fine-tuning. Given a target image (top) and a garment image (bottom), an image is generated using a diffusion-based prior that replaces the input garment with the one already worn in the target (right).

Abstract

Virtual Try-On (VTON) is a highly active line of research, with increasing demand. It aims to replace a piece of garment in an image with one from another, while preserving person and garment characteristics as well as image fidelity. Current literature takes a supervised approach for the task, impairing generalization and imposing heavy computation. In this paper, we present a novel zero-shot training-free method for inpainting a clothing garment by reference. Our approach employs the prior of a diffusion model with no additional training, fully leveraging its native generalization capabilities. The method employs extended attention to transfer image information from reference to target images, overcoming two significant challenges. We first initially warp the reference garment over the target human using deep features, alleviating "texture sticking". We then leverage the extended attention mechanism with careful masking, eliminating leakage of reference background and unwanted influence. Through a user study, qualitative, and quantitative comparison to state-of-the-art approaches, we demonstrate superior image quality and garment preservation compared unseen clothing pieces or human figures.

Results

How does it work?

Overview of MaX4Zero method

Top: The Initial Registration stage, where the reference garment is warped to match the target person using extracted deep features from both images. The remaining gaps between the target garment and warped one are filled by the Fringe Assignment module.

Bottom: The Consistent Inpainting stage, utilizing the Masked Extended Attention mechanism for transferring the reference fine-details through stroke-based inpainting.

Qualitative Comparisons

Qualitative comparison of MaX4Zero and competitors on in the wild target images. We compare against two dedicated VTON approaches (LaDI-VTON and StableVITON), personalized image editing approach (IP-adapter) and paint-by-reference approach (Anydoor). As can be seen, garment identity and image authenticity are preserved better using our method for unseen garments.

More Results

Results Using Out-of-Domain Reference Images

Results Using Out-of-Domain Target Images

BibTeX

@misc{orzech2024masked,
      title={Masked Extended Attention for Zero-Shot Virtual Try-On In The Wild}, 
      author={Nadav Orzech and Yotam Nitzan and Ulysse Mizrahi and Dov Danon and Amit H. Bermano},
      year={2024},
      eprint={2406.15331},
      archivePrefix={arXiv},
      primaryClass={id='cs.CV' full_name='Computer Vision and Pattern Recognition' is_active=True alt_name=None in_archive='cs' is_general=False description='Covers image processing, computer vision, pattern recognition, and scene understanding. Roughly includes material in ACM Subject Classes I.2.10, I.4, and I.5.'}
    }