CV-ZMH / VideoMatting

Background Removal in Video or Image with MODNet

Background Matting

Video matting with MODNet for video and image. Original repo is here.

Install Dependencies

• pytorch > 1.0

Install other third party packages with pip install -r requirements.txt

Run Demo

Download pretrained model from here. Then, run below command

python demo.py \
--source <input_path> \
--background <background_path> \
--weight <model_file_path> \
--demo_mode  

Citation

@article{MODNet,
  author = {Zhanghan Ke and Kaican Li and Yurou Zhou and Qiuhua Wu and Xiangyu Mao and Qiong Yan and Rynson W.H. Lau},
  title = {Is a Green Screen Really Necessary for Real-Time Portrait Matting?},
  journal={ArXiv},
  volume={abs/2011.11961},
  year = {2020},
}