Source-linked AI summary

SwiftTry: Fast and Consistent Video Virtual Try-On with Diffusion Models

Hung Nguyen, Quang Qui-Vinh Nguyen, Khoi Nguyen, Rang Nguyen

arXiv:2412.10178v2cs.CVcs.AI

TL;DR

Video virtual try-on must balance temporal coherence with the computational demands of long sequences. SwiftTry uses diffusion-based inpainting with temporal attention and ShiftCaching; on VVT, it reaches 2.27 FPS with ShiftCaching and excels in VFID.

  • Problem

    Video virtual try-on remains challenging because it must balance spatiotemporal consistency with the computational cost of processing long sequences, and existing datasets lack in-the-wild variety.

    Source: Introduction · p. 1
  • Method

    SwiftTry treats video virtual try-on as inpainting, adds temporal attention to model correlations across frames, and uses ShiftCaching to smooth chunk transitions while reducing redundant computation.

    Source: Introduction · p. 2, Overall Architecture · pp. 3–4
  • Results

    2.27 FPS—over 1.5 times faster with ShiftCaching; SwiftTry excels in VFID on VVT and performs competitively in SSIM and LPIPS.

    Source: Comparisons with Prior Approaches · p. 6
  • Takeaways & Limitations

    TikTokDress addresses limitations of existing video try-on datasets and provides a resource for future research with diverse backgrounds, complex movements, and high-resolution videos.

    Source: Conclusion · p. 7

    Training relies on self-supervision with a single video and masked garment regions because paired same-person, same-motion source and target videos are extremely challenging to collect.

    Source: Methods · p. 2

Abstract

from arXiv · show

Given an input video of a person and a new garment, the objective of this paper is to synthesize a new video where the person is wearing the specified garment while maintaining spatiotemporal consistency. Although significant advances have been made in image-based virtual try-on, extending these successes to video often leads to frame-to-frame inconsistencies. Some approaches have attempted to address this by increasing the overlap of frames across multiple video chunks, but this comes at a steep computational cost due to the repeated processing of the same frames, especially for long video sequences. To tackle these challenges, we reconceptualize video virtual try-on as a conditional video inpainting task, with garments serving as input conditions. Specifically, our approach enhances image diffusion models by incorporating temporal attention layers to improve temporal coherence. To reduce computational overhead, we propose ShiftCaching, a novel technique that maintains temporal consistency while minimizing redundant computations. Furthermore, we introduce the TikTokDress dataset, a new video try-on dataset featuring more complex backgrounds, challenging movements, and higher resolution compared to existing public datasets. Extensive experiments demonstrate that our approach outperforms current baselines, particularly in terms of video consistency and inference speed. The project page is available at https://swift-try.github.io/.

Introduction

Video virtual try-on must maintain temporal consistency while controlling the cost of processing long sequences, and existing datasets and methods leave important gaps. SwiftTry addresses these challenges with the TikTokDress dataset and a diffusion-based framework using temporal attention and ShiftCaching.

  • Over 60% faster runtime than ViViD (Fang et al. 2024) is reported for SwiftTry, which preserves garment texture detail and consistency.The introduction also states that SwiftTry outperforms existing video virtual try-on methods in accuracy and efficiency.Source: Figure 1 · p. 1, Introduction · p. 2
  • Video try-on methods can flicker between frames, while rendering high-quality results over long sequences remains computationally expensive.These limitations undermine realism and practical use.Source: Introduction · p. 1
  • Existing datasets lack the diverse in-the-wild scenes and movements needed for evaluation, and inaccurate human-parsing masks can degrade results.VVT has basic garments, uniform backgrounds, static camera angles, and repetitive motions; ViViD still struggles with complex movements and diverse backgrounds.Source: Introduction · p. 1
  • TikTokDress provides 817 videos with realistic scenes, diverse garments, complex movements, varied backgrounds, and high resolution for training and evaluating video try-on models.Source: Introduction · p. 2
  • SwiftTry adds temporal attention to a diffusion-based image virtual try-on UNet and uses ShiftCaching to smooth transitions between video chunks while reducing redundant computation.ShiftCaching is applied at inference, while temporal attention is incorporated for training on video try-on data.Source: Introduction · p. 2

Related Work

Prior image try-on methods use GAN-based warping or diffusion conditioning, while video methods address temporal coherence but remain limited by inconsistency, short demonstrations, or simple data. SwiftTry proposes temporal smoothing across chunks and caching to reduce redundant computation.

Image Virtual Try-On

  • Traditional image try-on commonly uses a two-stage GAN pipeline that warps clothing before fusing it with the person image, but can generalize poorly to complex backgrounds.Warping approaches include thin-plate splines, spatial transformer networks, and flow estimation.Source: Related Work · p. 2
  • Diffusion-based image try-on methods improve realism, but extending them to video can cause inter-frame inconsistency and flickering.TryOnDiffusion uses two U-Nets and requires many same-person, different-pose image pairs; StableVITON uses zero cross-attention for garment conditioning, while IDM-VTON combines low-level GarmentNet features with IP-Adapter semantics.Source: Related Work · p. 2

Video Virtual Try-On

  • Video try-on methods use optical flow, memory refinement, transformers, or diffusion, while prior work has limits in video length or dataset complexity.FW-GAN warps preceding frames, MV-TON refines prior-frame features, and ClothFormer combines TPS- and appearance-based warping; Tunnel Try-On handles camera movement but its demos span only a few seconds, while ViViD has simple backgrounds and movements. SwiftTry proposes temporal smoothness across chunks and caching to reduce redundant computation during long-video inference.Source: Related Work · p. 2

Methods

SwiftTry treats video virtual try-on as garment-conditioned inpainting, extending an image diffusion model with temporal attention and ShiftCaching to preserve appearance and motion while reducing redundant computation. It trains with masked garment regions because paired videos of the same person in identical motion are difficult to collect.

Methods

  • SwiftTry aims to generate a garment-conditioned video that preserves the person’s motion, source background, and garment color and texture.Because paired source and target videos with identical motion are difficult to collect, training uses a single video with garment regions masked and inpainted using the garment image as guidance.Source: Methods · p. 2

Overall Architecture

  • SwiftTry trains an image try-on diffusion model, then adapts its Main U-Net for video with temporal attention layers.The image model uses a 13-channel input combining latent noise, masked-image latents, a binary mask, and a pose map; a Garment U-Net supplies garment features through Reference Attention.Source: Overall Architecture · pp. 2–3, Figure 2 · p. 3
  • Temporal attention operates across frames at matching latent locations, factorized after spatial attention to avoid full 3D attention; training updates only the temporal layers.The model adds sinusoidal frame-position encodings and trains the temporal layers on video data while keeping other layers unchanged.Source: Overall Architecture · pp. 3–4

ShiftCaching Technique

  • ShiftCaching shifts non-overlapping video chunks between DDIM steps and uses cached features for partial Main U-Net computation to avoid recomputing overlapping regions.Previous overlap-based methods trade speed for consistency: small overlaps can cause flickering and texture artifacts, while large overlaps slow inference. Masked Temporal Attention blocks information flow from less accurate cached features to more accurate ones while allowing the reverse, supporting smoothness and quality.Source: ShiftCaching Technique · p. 4

TikTokDress Dataset

TikTokDress addresses limitations in existing virtual try-on datasets with high-resolution videos, diverse real-world conditions, accurate garment masks, and detailed annotations.

TikTokDress Dataset

  • TikTokDress pairs high-resolution videos with garments, poses, and precise binary cloth masks, targeting complex backgrounds, diverse movements, and balanced gender representation.The dataset is intended to address limitations of single-image datasets and VVT, including restricted poses, uniform movement, white backgrounds, and low resolution.Source: TikTokDress Dataset · pp. 4–5
  • Manually corrected garment masks improve video try-on quality, while standard segmenters used by VITON-HD, DressCode, and VVT struggle with complex videos.Mask quality is described as crucial for try-on results; the cited comparison is qualitative.Source: TikTokDress Dataset · p. 5
  • TikTokDress covers dancing and other dynamic poses, varied camera distances, indoor and outdoor backgrounds with complex lighting, and garments from T-shirts to chainmail tops.The range is intended to address real-world challenges in video virtual try-on.Source: TikTokDress Dataset · p. 5

Video collection and annotation.

  • The dataset contains 817 video-garment pairs and over 270,000 RGB frames sampled at 30 frames per second, with 2D keypoints and dense poses computed using DWPose and DensePose.Videos are 10–30-second TikTok clips; the collection was expanded for background, skin-tone, and clothing diversity, and low-quality videos or garments were excluded.Source: Video collection and annotation · p. 5
  • SAM 2 masks for clothing and arms required an algorithm for selecting frames and prompts, plus manual refinement for complex garments (Ravi et al. 2024).The procedure addresses SAM 2’s sensitivity to prompt points and particular frames.Source: Video collection and annotation · p. 5

Experiments

SwiftTry is evaluated on VVT and TikTokDress, with results indicating strong temporal consistency and faster inference; ablations identify effective ShiftCaching, attention-mask, and chunk-length settings.

Experiments

  • The evaluation uses VVT and TikTokDress with paired SSIM and LPIPS, unpaired VFID, and FPS; training first adapts garment and inpainting modules, then adds temporal attention.VVT has 791 paired videos at 256 × 192 resolution; TikTokDress has 693 training and 124 testing videos at 540 × 720. Temporal attention is initialized from AnimateDiff weights.Source: Experiments · pp. 5–6

Comparisons with Prior Approaches

Ablation Study

  • Random shifts provide the best consistency; partially computing 50% of frames accelerates inference 1.5 times while maintaining comparable quantitative metrics.This is the ShiftCaching ablation on VVT.Source: Ablation Study · p. 6, Table 5 · p. 6
  • Half Attention performs best, allowing bad features from timestep t + 2 to access good features from timestep t + 1 while good features interact with each other.The study examines masking choices for temporal attention on VVT.Source: Table 6 · p. 6, Ablation Study · p. 7
  • Matching training and inference video chunk lengths at N = 16 yields the best results.The study compares inference chunks of 8, 16, and 24 frames with training set to 16 frames.Source: Table 7 · p. 6, Ablation Study · p. 7

Conclusion

The conclusion presents ShiftCaching as a way to maintain temporal smoothness across video chunks while reducing redundant inference computation, and introduces TikTokDress as a dataset with diverse backgrounds, complex movements, and high-resolution videos.

  • ShiftCaching maintains temporal smoothness across video chunks while reducing redundant computations during video inference.The paper says this improves the efficiency and quality of video virtual try-on and makes it more practical for real-world applications.Source: Conclusion · p. 7
  • TikTokDress is a video virtual try-on dataset featuring diverse backgrounds, complex movements, and high-resolution videos.It is designed to address limitations of existing datasets and support future research.Source: Conclusion · p. 7

Supplementary Material

The supplementary material expands on SwiftTry’s training process, attention masking strategies, TikTokDress dataset creation, and qualitative results on VVT and TikTokDress.

Training Detail

Training proceeds in two stages, first optimizing image-related modules on VITON-HD and then motion-related modules on sampled video clips. Both stages use the same optimizer settings and run on a single A100 GPU.

Optimization

Hardware

  • Training runs in PyTorch with Diffusers (von Platen et al. 2022) on one 40GB A100; stage 1 takes approximately 3 days and stage 2 around 30 hours.FPS benchmarks are also measured on the A100 GPU.Source: Training Detail · p. 10

Training

  • Stage 1 optimizes the Main UNet decoder, Garment UNet, and Pose Encoder on 1024×768 VITON-HD; stage 2 optimizes only motion-related modules using 16-frame clips at 512 × 384.All other modules are fixed in stage 1.Source: Training Detail · p. 10

Different Masking Attention Mechanisms

ShiftCaching tests four temporal-attention masks, with Half Attention performing best by directing query frames to accurate features.

TikTokDress Dataset

TikTokDress includes diverse people, garments, and settings, and its try-on data preparation emphasizes accurate agnostic masks and garment images.

TikTokDress Dataset

  • The dataset spans indoor and outdoor backgrounds, varied camera angles and person distances, Black, Asian, and White skin tones, and multiple garment types.Examples include T-shirts, sweaters, crop tops, long-sleeve T-shirts, and dresses.Source: TikTokDress Dataset · p. 10
  • Incorrect or incomplete garment masks can produce hallucinated clothing shapes or stripes, while precise masks yield more accurate try-on results.The examples show both a tank-top-like output from an incorrect mask and extra red stripes when the garment is not fully masked.Source: TikTokDress Dataset · p. 10, Figure 10 · p. 11
  • Agnostic masks combine SAM2 garment and arm segmentations from a clear frame with the most visible body parts, then are dilated three to seven times.Garment prompts use body-center and shoulder positives; arm prompts use elbow and wrist positives, with specified negative points for both. Frame selection ranks visible parts first, then pose score.Source: TikTokDress Dataset · pp. 10–11
  • Garment images are sourced through Google Lens at the highest available resolution, with manual keyword searches as fallback; videos are excluded if no suitable image is found.Manual searches use garment type, color, and brand keywords.Source: TikTokDress Dataset · p. 11

More Visualization Results

Qualitative results show that SwiftTry preserves garment textures and produces temporally coherent try-on videos on both VVT and TikTokDress. The TikTokDress results also demonstrate performance across varied backgrounds, supported by refined masks and SAM 2 frame selection.

Qualitative results on the VVT dataset

  • On VVT, SwiftTry preserves garment textures and reduces flickering relative to ViViD, including flicker in the garment’s “GAP” lettering caused by ViViD’s temporal averaging.The authors attribute smoother, more stable results to ShiftCaching.Source: Qualitative results on the VVT dataset · p. 11, Figure 13 · p. 12

Qualitative results on the TikTokDress dataset.

Loading 2412.10178v2…