Skip to content

hufangjian/BackgroundMattingV2

 
 

Repository files navigation

Real-Time High-Resolution Background Matting

Teaser

Official repository for the paper Real-Time High-Resolution Background Matting. Our model requires capturing an additional background image and produces state-of-the-art matting results at 4K 30fps and HD 60fps on an Nvidia RTX 2080 TI GPU.

 

Overview

 

Download

Model / Weights

Video / Image Examples

Datasets

  • VideoMatte240K (Coming soon)
  • PhotoMatte85 (Coming soon)

 

Demo

Scripts

We provide several scripts in this repo for you to experiment with our model. More detailed instructions are included in the files.

  • inference_images.py: Perform matting on a directory of images.
  • inference_video.py: Perform matting on a video.
  • inference_webcam.py: An interactive matting demo using your webcam.

Notebooks

Additionally, you can try our notebooks in Google Colab for performing matting on images and videos.

Virtual Camera

We provide a demo application that pipes webcam video through our model and outputs to a virtual camera. The script only works on Linux system and can be used in Zoom meetings. For more information, checkout:

Web Demo

Developers in the community has helped us build a web demo. See Community Projects section below.

 

Usage / Documentation

You can run our model using PyTorch, TorchScript, TensorFlow, and ONNX. For detail about using our model, please check out the Usage / Documentation page.

 

Training

Training code will be released upon acceptance of the paper.

 

Project members

* Equal contribution.

 

License

This work is licensed under the Creative Commons Attribution NonCommercial ShareAlike 4.0 License.

 

Community Projects

A list of projects built by third-party developers in the community. If you have a project to share, contact us!

About

Real-Time High-Resolution Background Matting

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%