├── LICENSE ├── README.md ├── alpha2trimap.py ├── configs ├── common │ ├── dataloader.py │ ├── model.py │ ├── optimizer.py │ ├── resnet_model.py │ ├── scheduler.py │ └── train.py └── vitmatte_s_am2k_100ep.py ├── data ├── __init__.py ├── am2k_dataset.py └── evaluation.py ├── demo.sh ├── docs ├── installation.md ├── test.md └── train.md ├── engine ├── __init__.py └── mattingtrainer.py ├── evaluation.py ├── evaluation_whole.py ├── inference.py ├── main.py ├── modeling ├── __init__.py ├── backbone │ ├── __init__.py │ ├── backbone.py │ ├── utils.py │ └── vit.py ├── criterion │ ├── __init__.py │ └── matting_criterion.py ├── decoder │ ├── __init__.py │ └── detail_capture.py └── meta_arch │ ├── __init__.py │ └── vitmatte.py ├── pretrained └── preprocess.py ├── requirements.txt ├── run_one_image.py └── train.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/README.md -------------------------------------------------------------------------------- /alpha2trimap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/alpha2trimap.py -------------------------------------------------------------------------------- /configs/common/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/common/dataloader.py -------------------------------------------------------------------------------- /configs/common/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/common/model.py -------------------------------------------------------------------------------- /configs/common/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/common/optimizer.py -------------------------------------------------------------------------------- /configs/common/resnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/common/resnet_model.py -------------------------------------------------------------------------------- /configs/common/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/common/scheduler.py -------------------------------------------------------------------------------- /configs/common/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/common/train.py -------------------------------------------------------------------------------- /configs/vitmatte_s_am2k_100ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/configs/vitmatte_s_am2k_100ep.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/am2k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/data/am2k_dataset.py -------------------------------------------------------------------------------- /data/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/data/evaluation.py -------------------------------------------------------------------------------- /demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/demo.sh -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/docs/installation.md -------------------------------------------------------------------------------- /docs/test.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/docs/test.md -------------------------------------------------------------------------------- /docs/train.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/docs/train.md -------------------------------------------------------------------------------- /engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/engine/__init__.py -------------------------------------------------------------------------------- /engine/mattingtrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/engine/mattingtrainer.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/evaluation.py -------------------------------------------------------------------------------- /evaluation_whole.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/evaluation_whole.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/inference.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/main.py -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/__init__.py -------------------------------------------------------------------------------- /modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /modeling/backbone/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/backbone/backbone.py -------------------------------------------------------------------------------- /modeling/backbone/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/backbone/utils.py -------------------------------------------------------------------------------- /modeling/backbone/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/backbone/vit.py -------------------------------------------------------------------------------- /modeling/criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/criterion/__init__.py -------------------------------------------------------------------------------- /modeling/criterion/matting_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/criterion/matting_criterion.py -------------------------------------------------------------------------------- /modeling/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/decoder/__init__.py -------------------------------------------------------------------------------- /modeling/decoder/detail_capture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/decoder/detail_capture.py -------------------------------------------------------------------------------- /modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/meta_arch/__init__.py -------------------------------------------------------------------------------- /modeling/meta_arch/vitmatte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/modeling/meta_arch/vitmatte.py -------------------------------------------------------------------------------- /pretrained/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/pretrained/preprocess.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_one_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/run_one_image.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/poppuppy/alpha-free-matting/HEAD/train.sh --------------------------------------------------------------------------------