├── LICENSE ├── README.md ├── README_zh_Hans.md ├── dataset ├── augmentation.py ├── coco.py ├── imagematte.py ├── spd.py ├── videomatte.py └── youtubevis.py ├── documentation ├── image │ ├── showreel.gif │ └── teaser.gif ├── inference.md ├── inference_zh_Hans.md ├── misc │ ├── aim_test.txt │ ├── d646_test.txt │ ├── dvm_background_test_clips.txt │ ├── dvm_background_train_clips.txt │ ├── imagematte_train.txt │ ├── imagematte_valid.txt │ └── spd_preprocess.py └── training.md ├── evaluation ├── evaluate_hr.py ├── evaluate_lr.py ├── generate_imagematte_with_background_image.py ├── generate_imagematte_with_background_video.py ├── generate_videomatte_with_background_image.py └── generate_videomatte_with_background_video.py ├── hubconf.py ├── inference.py ├── inference_speed_test.py ├── inference_utils.py ├── model ├── __init__.py ├── decoder.py ├── deep_guided_filter.py ├── fast_guided_filter.py ├── lraspp.py ├── mobilenetv3.py ├── model.py └── resnet.py ├── requirements_inference.txt ├── requirements_training.txt ├── train.py ├── train_config.py └── train_loss.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/README.md -------------------------------------------------------------------------------- /README_zh_Hans.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/README_zh_Hans.md -------------------------------------------------------------------------------- /dataset/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/dataset/augmentation.py -------------------------------------------------------------------------------- /dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/dataset/coco.py -------------------------------------------------------------------------------- /dataset/imagematte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/dataset/imagematte.py -------------------------------------------------------------------------------- /dataset/spd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/dataset/spd.py -------------------------------------------------------------------------------- /dataset/videomatte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/dataset/videomatte.py -------------------------------------------------------------------------------- /dataset/youtubevis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/dataset/youtubevis.py -------------------------------------------------------------------------------- /documentation/image/showreel.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/image/showreel.gif -------------------------------------------------------------------------------- /documentation/image/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/image/teaser.gif -------------------------------------------------------------------------------- /documentation/inference.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/inference.md -------------------------------------------------------------------------------- /documentation/inference_zh_Hans.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/inference_zh_Hans.md -------------------------------------------------------------------------------- /documentation/misc/aim_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/aim_test.txt -------------------------------------------------------------------------------- /documentation/misc/d646_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/d646_test.txt -------------------------------------------------------------------------------- /documentation/misc/dvm_background_test_clips.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/dvm_background_test_clips.txt -------------------------------------------------------------------------------- /documentation/misc/dvm_background_train_clips.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/dvm_background_train_clips.txt -------------------------------------------------------------------------------- /documentation/misc/imagematte_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/imagematte_train.txt -------------------------------------------------------------------------------- /documentation/misc/imagematte_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/imagematte_valid.txt -------------------------------------------------------------------------------- /documentation/misc/spd_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/misc/spd_preprocess.py -------------------------------------------------------------------------------- /documentation/training.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/documentation/training.md -------------------------------------------------------------------------------- /evaluation/evaluate_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/evaluation/evaluate_hr.py -------------------------------------------------------------------------------- /evaluation/evaluate_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/evaluation/evaluate_lr.py -------------------------------------------------------------------------------- /evaluation/generate_imagematte_with_background_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/evaluation/generate_imagematte_with_background_image.py -------------------------------------------------------------------------------- /evaluation/generate_imagematte_with_background_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/evaluation/generate_imagematte_with_background_video.py -------------------------------------------------------------------------------- /evaluation/generate_videomatte_with_background_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/evaluation/generate_videomatte_with_background_image.py -------------------------------------------------------------------------------- /evaluation/generate_videomatte_with_background_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/evaluation/generate_videomatte_with_background_video.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/hubconf.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/inference.py -------------------------------------------------------------------------------- /inference_speed_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/inference_speed_test.py -------------------------------------------------------------------------------- /inference_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/inference_utils.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import MattingNetwork -------------------------------------------------------------------------------- /model/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/decoder.py -------------------------------------------------------------------------------- /model/deep_guided_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/deep_guided_filter.py -------------------------------------------------------------------------------- /model/fast_guided_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/fast_guided_filter.py -------------------------------------------------------------------------------- /model/lraspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/lraspp.py -------------------------------------------------------------------------------- /model/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/mobilenetv3.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/model.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/model/resnet.py -------------------------------------------------------------------------------- /requirements_inference.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/requirements_inference.txt -------------------------------------------------------------------------------- /requirements_training.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/requirements_training.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/train.py -------------------------------------------------------------------------------- /train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/train_config.py -------------------------------------------------------------------------------- /train_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/skywalk163/RobustVideoMatting/HEAD/train_loss.py --------------------------------------------------------------------------------