EffectErase

 ·   ·   ·  Institute of Big Data, College of Computer Science and Artificial Intelligence, Fudan University, China

This repository provides the checkpoint EffectErase.ckpt for EffectErase.

Abstract

Video object removal aims to eliminate dynamic target objects and their visual effects, such as deformation, shadows, and reflections, while restoring seamless backgrounds. Recent diffusion-based video inpainting and object removal methods can remove the objects but often struggle to erase these effects and to synthesize coherent backgrounds. Beyond method limitations, progress is further hampered by the lack of a comprehensive dataset that systematically captures common object effects across varied environments for training and evaluation. To address this, we introduce VOR (Video Object Removal), a large-scale dataset that provides diverse paired videos, each consisting of one video where the target object is present with its effects and a counterpart where the object and effects are absent, with corresponding object masks. VOR contains 60k high-quality video pairs from captured and synthetic sources, covers five effects types, and spans a wide range of object categories as well as complex, dynamic multi-object scenes. Building on VOR, we propose EffectErase, an effect-aware video object removal method that treats video object insertion as the inverse auxiliary task within a reciprocal learning scheme. The model includes task-aware region guidance that focuses learning on affected areas and enables flexible task switching. Then, an insertion–removal consistency objective that encourages complementary behaviors and shared localization of effect regions and structural cues. Trained on VOR, EffectErase achieves superior performance in extensive experiments, delivering high-quality video object effect erasing across diverse scenarios.

Quick Start

1. Setup repository and environment

   git clone git@github.com:FudanCVL/EffectErase.git
   cd EffectErase
   pip install -e .
   

2. Download weights

   hf download alibaba-pai/Wan2.1-Fun-1.3B-InP --local-dir Wan-AI/Wan2.1-Fun-1.3B-InP
   hf download FudanCVL/EffectErase EffectErase.ckpt --local-dir ./
   

3. Run the script

   bash script/test_remove.sh
   

   You can edit script/test_remove.sh and change these three paths to use your own data:

   • --fg_bg_path
   • --mask_path
   • --output_path

   --mask_path is a mask video generated by SAM2.1 (sam2.1_hiera_b+), aligned with --fg_bg_path.

BibTeX

Please consider to cite:

@inproceedings{fu2026effecterase,
  title={EffectErase: Joint Video Object Removal and Insertion for High-Quality Effect Erasing},
  author={Fu, Yang and Zheng, Yike and Dai, Ziyun and Ding, Henghui},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  year={2026}
}

Contact

If you have any questions, please feel free to reach me out at aleeyanger@gmail.com.