Update README.md

README.md

@@ -1,4 +1,66 @@
 ---
-license: cc-by-nc-4.0
+license:  cc-by-nc-4.0
 pipeline_tag: video-to-video
----
+library_name: diffusers
+---
+
+# EffectErase: Joint Video Object Removal and Insertion for High-Quality Effect Erasing
+
+[![Project Page](https://img.shields.io/badge/🐳-Project%20Page-blue)](https://henghuiding.com/EffectErase/) [![Paper](https://img.shields.io/badge/Paper-CVPR%202026-green)](https://cvpr.thecvf.com/virtual/2026/papers.html) [![GitHub](https://img.shields.io/badge/GitHub-FudanCVL%2FEffectErase-181717?logo=github)](https://github.com/FudanCVL/EffectErase) [![arXiv](https://img.shields.io/badge/arXiv-EffectErase-red)](http://arxiv.org/) [![Dataset](https://img.shields.io/badge/%F0%9F%A4%97%20Dataset-Hugging%20Face-yellow)](https://huggingface.co/datasets/FudanCVL/EffectErase_VOR)
+
+This repository provides the checkpoint `EffectErase.ckpt` for **EffectErase**.
+
+<img src="assets/teaser.gif" alt="teaser" />
+
+## 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** (**V**ideo **O**bject **R**emoval), 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
+
+   ```bash
+   git clone git@github.com:FudanCVL/EffectErase.git
+   cd EffectErase
+   pip install -e .
+   ```
+
+2. Download weights
+
+   ```bash
+   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
+   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:
+
+```bibtex
+@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.
+