Skip to content

Reagan1311/Mask2IV

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Mask2IV: Interaction-Centric Video Generation via Mask Trajectories

arXiv GitHub

Abstract

Generating interaction-centric videos, such as those depicting humans or robots interacting with objects, is crucial for embodied intelligence, as they provide rich and diverse visual priors for robot learning, manipulation policy training, and affordance reasoning. However, existing methods often struggle to model such complex and dynamic interactions. While recent studies show that masks can serve as effective control signals and enhance generation quality, obtaining dense and precise mask annotations remains a major challenge for real-world use. To overcome this limitation, we introduce Mask2IV, a novel framework specifically designed for interaction-centric video generation. It adopts a decoupled two-stage pipeline that first predicts plausible motion trajectories for both actor and object, then generates a video conditioned on these trajectories. This design eliminates the need for dense mask inputs from users while preserving the flexibility to manipulate the interaction process. Furthermore, Mask2IV supports versatile and intuitive control, allowing users to specify the target object of interaction and guide the motion trajectory through action descriptions or spatial position cues. To support systematic training and evaluation, we curate two benchmarks covering diverse action and object categories across both human-object interaction and robotic manipulation scenarios. Extensive experiments demonstrate that our method achieves superior visual realism and controllability compared to existing baselines.

Setup

1. Environment Installation

Install environment following DynamiCrafter.

2. Datasets Preparation

(1) Download datasets of HOI4D and BridgeDataV2

(2) Update data_dir in config files

(3) Update meta_path (data csv files, in /data_csv folder) in config files

PS: Please refer to scripts in /func folder for details on data preprocessing.

Training & Evaluation

Download pretrained models DynamiCrafter_512 and DynamiCrafter512_interp, and update pretrained_checkpoint in config files.

I. Stage 1 training & evaluation

sh run_fist.sh hoi4d | bdv2

II. Stage 2 training & evaluation

sh run_second.sh hoi4d | bdv2

Pretrained models of Mask2IV can be downloaded here: Huggingface

Inference

sh scripts/run_hoi4d_2tage.sh  # HOI4D
sh scripts/run_bdv2_2stage.sh  # BridgeDataV2 

Citation

@inproceedings{li2026mask2iv,
      title     = {Mask2IV: Interaction-Centric Video Generation via Mask Trajectories}, 
      author    = {Li, Gen and Zhao, Bo and Yang, Jianfei and Sevilla-Lara, Laura},
      journal   = {Proceedings of the AAAI Conference on Artificial Intelligence},
      year      = {2026},
    }

Anckowledgement

A large portion of the code is based on DynamiCrafter. Thanks for their great work!

About

Mask2IV: Interaction-Centric Video Generation via Mask Trajectories (AAAI 2026)

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published