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
-
Setup repository and environment
git clone git@github.com:FudanCVL/EffectErase.git cd EffectErase pip install -e . -
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 ./ -
Run the script
bash script/test_remove.shYou can edit
script/test_remove.shand change these three paths to use your own data:--fg_bg_path--mask_path--output_path
--mask_pathis 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.