├── .gitmodules ├── LICENSE ├── common ├── __init__.py ├── image_helper.py ├── lr_scheduler.py ├── model_loader.py ├── tf_utils.py └── utils.py ├── const.py ├── datasets ├── __init__.py ├── augmentation_factory.py ├── data_wrapper_base.py └── matting_data_wrapper.py ├── demo └── demo.mp4 ├── evaluate.py ├── factory ├── __init__.py ├── base.py ├── losses_funcs.py ├── matting_converter.py └── matting_nets.py ├── figure └── gradient_error_vs_latency.png ├── helper ├── __init__.py ├── base.py ├── evaluator.py └── trainer.py ├── matting_nets ├── __init__.py ├── deeplab_v3plus.py ├── mmnet.py └── mmnet_utils.py ├── metrics ├── base.py ├── manager.py ├── ops │ ├── __init__.py │ ├── base_ops.py │ ├── misc_ops.py │ └── tensor_ops.py ├── parser.py └── summaries.py ├── nets ├── __init__.py └── mobilenet ├── readme.md ├── requirements ├── py36-common.txt ├── py36-cpu.txt └── py36-gpu.txt ├── scripts ├── train_deeplab_os16_dm0.5_256.sh ├── train_mmnet_dm1.0_256.sh ├── valid_deeplab_os16_dm0.5_256.sh └── valid_mmnet_dm1.0_256.sh └── train.py /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/LICENSE -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/image_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/common/image_helper.py -------------------------------------------------------------------------------- /common/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/common/lr_scheduler.py -------------------------------------------------------------------------------- /common/model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/common/model_loader.py -------------------------------------------------------------------------------- /common/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/common/tf_utils.py -------------------------------------------------------------------------------- /common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/common/utils.py -------------------------------------------------------------------------------- /const.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/const.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/augmentation_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/datasets/augmentation_factory.py -------------------------------------------------------------------------------- /datasets/data_wrapper_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/datasets/data_wrapper_base.py -------------------------------------------------------------------------------- /datasets/matting_data_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/datasets/matting_data_wrapper.py -------------------------------------------------------------------------------- /demo/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/demo/demo.mp4 -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/evaluate.py -------------------------------------------------------------------------------- /factory/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /factory/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/factory/base.py -------------------------------------------------------------------------------- /factory/losses_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/factory/losses_funcs.py -------------------------------------------------------------------------------- /factory/matting_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/factory/matting_converter.py -------------------------------------------------------------------------------- /factory/matting_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/factory/matting_nets.py -------------------------------------------------------------------------------- /figure/gradient_error_vs_latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/figure/gradient_error_vs_latency.png -------------------------------------------------------------------------------- /helper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /helper/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/helper/base.py -------------------------------------------------------------------------------- /helper/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/helper/evaluator.py -------------------------------------------------------------------------------- /helper/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/helper/trainer.py -------------------------------------------------------------------------------- /matting_nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /matting_nets/deeplab_v3plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/matting_nets/deeplab_v3plus.py -------------------------------------------------------------------------------- /matting_nets/mmnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/matting_nets/mmnet.py -------------------------------------------------------------------------------- /matting_nets/mmnet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/matting_nets/mmnet_utils.py -------------------------------------------------------------------------------- /metrics/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/base.py -------------------------------------------------------------------------------- /metrics/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/manager.py -------------------------------------------------------------------------------- /metrics/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/ops/__init__.py -------------------------------------------------------------------------------- /metrics/ops/base_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/ops/base_ops.py -------------------------------------------------------------------------------- /metrics/ops/misc_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/ops/misc_ops.py -------------------------------------------------------------------------------- /metrics/ops/tensor_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/ops/tensor_ops.py -------------------------------------------------------------------------------- /metrics/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/parser.py -------------------------------------------------------------------------------- /metrics/summaries.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/metrics/summaries.py -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nets/mobilenet: -------------------------------------------------------------------------------- 1 | ../models/research/slim/nets/mobilenet -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/readme.md -------------------------------------------------------------------------------- /requirements/py36-common.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/requirements/py36-common.txt -------------------------------------------------------------------------------- /requirements/py36-cpu.txt: -------------------------------------------------------------------------------- 1 | # Deep learning 2 | tensorflow==1.6.0 3 | -r py36-common.txt -------------------------------------------------------------------------------- /requirements/py36-gpu.txt: -------------------------------------------------------------------------------- 1 | # Deep learning 2 | tensorflow-gpu==1.6.0 3 | -r py36-common.txt -------------------------------------------------------------------------------- /scripts/train_deeplab_os16_dm0.5_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/scripts/train_deeplab_os16_dm0.5_256.sh -------------------------------------------------------------------------------- /scripts/train_mmnet_dm1.0_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/scripts/train_mmnet_dm1.0_256.sh -------------------------------------------------------------------------------- /scripts/valid_deeplab_os16_dm0.5_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/scripts/valid_deeplab_os16_dm0.5_256.sh -------------------------------------------------------------------------------- /scripts/valid_mmnet_dm1.0_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/scripts/valid_mmnet_dm1.0_256.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyperconnect/MMNet/HEAD/train.py --------------------------------------------------------------------------------