subDesTagesMitExtraKaese
77f1bc25c2
added webcam inference with compositing
|
4 gadi atpakaļ | |
|---|---|---|
| dataset | 4 gadi atpakaļ | |
| doc | 4 gadi atpakaļ | |
| images | 4 gadi atpakaļ | |
| model | 4 gadi atpakaļ | |
| LICENSE | 4 gadi atpakaļ | |
| README.md | 4 gadi atpakaļ | |
| data_path.py | 4 gadi atpakaļ | |
| export_onnx.py | 4 gadi atpakaļ | |
| export_torchscript.py | 4 gadi atpakaļ | |
| inference_images.py | 4 gadi atpakaļ | |
| inference_speed_test.py | 4 gadi atpakaļ | |
| inference_utils.py | 4 gadi atpakaļ | |
| inference_video.py | 4 gadi atpakaļ | |
| inference_webcam.py | 4 gadi atpakaļ | |
| inference_webcam_ts.py | 4 gadi atpakaļ | |
| inference_webcam_ts_compositing.py | 4 gadi atpakaļ | |
| requirements.txt | 4 gadi atpakaļ | |
| run.sh | 4 gadi atpakaļ | |
| runCompositing.sh | 4 gadi atpakaļ | |
| runTS.sh | 4 gadi atpakaļ | |
| train_base.py | 4 gadi atpakaļ | |
| train_refine.py | 4 gadi atpakaļ |
README.md
Real-Time High-Resolution Background Matting
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
Updates
- [Mar 06 2021] Training script is published.
- [Feb 28 2021] Paper is accepted to CVPR 2021.
- [Jan 09 2021] PhotoMatte85 dataset is now published.
- [Dec 21 2020] We updated our project to MIT License, which permits commercial use.
Download
Model / Weights
Video / Image Examples
- HD videos (by Sengupta et al.) (Our model is more robust on HD footage)
- 4K videos and images
Datasets
- PhotoMatte85
- VideoMatte240K (We are still dealing with licensing. In the meantime, you can visit storyblocks.com to download raw green screen videos and recreate the dataset yourself.)
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
Configure data_path.pth to point to your dataset. The original paper uses train_base.pth to train only the base model till convergence then use train_refine.pth to train the entire network end-to-end. More details are specified in the paper.
Project members
- Shanchuan Lin*, University of Washington
- Andrey Ryabtsev*, University of Washington
- Soumyadip Sengupta, University of Washington
- Brian Curless, University of Washington
- Steve Seitz, University of Washington
- Ira Kemelmacher-Shlizerman, University of Washington
* Equal contribution.
License
This work is licensed under the MIT License. If you use our work in your project, we would love you to include an acknowledgement and fill out our survey.
Community Projects
A list of projects built by third-party developers in the community. If you have a project to share, fill out this survey.
- Web Demo by Gradio: Matting your own images from your browser.