├── .gitignore ├── LICENSE ├── README.md ├── demo ├── image_matting │ └── colab │ │ ├── README.md │ │ └── inference.py └── video_matting │ ├── custom │ ├── README.md │ ├── requirements.txt │ └── run.py │ └── webcam │ ├── README.md │ ├── requirements.txt │ └── run.py ├── doc └── gif │ ├── commercial_image_matting_model_result.gif │ ├── commercial_image_matting_website.gif │ ├── homepage_demo.gif │ ├── image_matting_demo.gif │ └── video_matting_demo.gif ├── onnx ├── README.md ├── __init__.py ├── export_onnx.py ├── inference_onnx.py ├── modnet_onnx.py └── requirements.txt ├── pretrained └── README.md ├── src ├── __init__.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── mobilenetv2.py │ │ └── wrapper.py │ └── modnet.py └── trainer.py └── torchscript ├── README.md ├── __init__.py ├── export_torchscript.py └── modnet_torchscript.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/README.md -------------------------------------------------------------------------------- /demo/image_matting/colab/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/image_matting/colab/README.md -------------------------------------------------------------------------------- /demo/image_matting/colab/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/image_matting/colab/inference.py -------------------------------------------------------------------------------- /demo/video_matting/custom/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/video_matting/custom/README.md -------------------------------------------------------------------------------- /demo/video_matting/custom/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/video_matting/custom/requirements.txt -------------------------------------------------------------------------------- /demo/video_matting/custom/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/video_matting/custom/run.py -------------------------------------------------------------------------------- /demo/video_matting/webcam/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/video_matting/webcam/README.md -------------------------------------------------------------------------------- /demo/video_matting/webcam/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/video_matting/webcam/requirements.txt -------------------------------------------------------------------------------- /demo/video_matting/webcam/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/demo/video_matting/webcam/run.py -------------------------------------------------------------------------------- /doc/gif/commercial_image_matting_model_result.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/doc/gif/commercial_image_matting_model_result.gif -------------------------------------------------------------------------------- /doc/gif/commercial_image_matting_website.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/doc/gif/commercial_image_matting_website.gif -------------------------------------------------------------------------------- /doc/gif/homepage_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/doc/gif/homepage_demo.gif -------------------------------------------------------------------------------- /doc/gif/image_matting_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/doc/gif/image_matting_demo.gif -------------------------------------------------------------------------------- /doc/gif/video_matting_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/doc/gif/video_matting_demo.gif -------------------------------------------------------------------------------- /onnx/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/onnx/README.md -------------------------------------------------------------------------------- /onnx/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /onnx/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/onnx/export_onnx.py -------------------------------------------------------------------------------- /onnx/inference_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/onnx/inference_onnx.py -------------------------------------------------------------------------------- /onnx/modnet_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/onnx/modnet_onnx.py -------------------------------------------------------------------------------- /onnx/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/onnx/requirements.txt -------------------------------------------------------------------------------- /pretrained/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/pretrained/README.md -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/src/models/backbones/__init__.py -------------------------------------------------------------------------------- /src/models/backbones/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/src/models/backbones/mobilenetv2.py -------------------------------------------------------------------------------- /src/models/backbones/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/src/models/backbones/wrapper.py -------------------------------------------------------------------------------- /src/models/modnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/src/models/modnet.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/src/trainer.py -------------------------------------------------------------------------------- /torchscript/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/torchscript/README.md -------------------------------------------------------------------------------- /torchscript/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchscript/export_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/torchscript/export_torchscript.py -------------------------------------------------------------------------------- /torchscript/modnet_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZHKKKe/MODNet/HEAD/torchscript/modnet_torchscript.py --------------------------------------------------------------------------------