├── README.md ├── data ├── __init__.py ├── retrieve.py └── utils.py ├── data_dim ├── composition.py ├── retrieve.py └── test │ ├── alpha │ ├── 16452523375_08591714cf_o.png │ └── girl-1467820_1280.png │ ├── bg │ ├── 000000239187.jpg │ └── 000000393412.jpg │ ├── fg │ ├── 16452523375_08591714cf_o.png │ └── girl-1467820_1280.png │ └── trimap │ ├── 16452523375_08591714cf_o_0.png │ └── girl-1467820_1280_0.png ├── dataset.py ├── model.py ├── model_deconv.py ├── model_paper.py ├── predict.py ├── predict_trimap.py ├── test ├── 1803151818-00000003.jpg ├── 1803151818-00000003.png ├── 1803151818-00000004.jpg ├── 1803151818-00000004.png ├── 1803250719-00000103.jpg └── 1803250719-00000103.png └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /data/retrieve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data/retrieve.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data/utils.py -------------------------------------------------------------------------------- /data_dim/composition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/composition.py -------------------------------------------------------------------------------- /data_dim/retrieve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/retrieve.py -------------------------------------------------------------------------------- /data_dim/test/alpha/16452523375_08591714cf_o.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/alpha/16452523375_08591714cf_o.png -------------------------------------------------------------------------------- /data_dim/test/alpha/girl-1467820_1280.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/alpha/girl-1467820_1280.png -------------------------------------------------------------------------------- /data_dim/test/bg/000000239187.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/bg/000000239187.jpg -------------------------------------------------------------------------------- /data_dim/test/bg/000000393412.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/bg/000000393412.jpg -------------------------------------------------------------------------------- /data_dim/test/fg/16452523375_08591714cf_o.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/fg/16452523375_08591714cf_o.png -------------------------------------------------------------------------------- /data_dim/test/fg/girl-1467820_1280.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/fg/girl-1467820_1280.png -------------------------------------------------------------------------------- /data_dim/test/trimap/16452523375_08591714cf_o_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/trimap/16452523375_08591714cf_o_0.png -------------------------------------------------------------------------------- /data_dim/test/trimap/girl-1467820_1280_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/data_dim/test/trimap/girl-1467820_1280_0.png -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/dataset.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/model.py -------------------------------------------------------------------------------- /model_deconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/model_deconv.py -------------------------------------------------------------------------------- /model_paper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/model_paper.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/predict.py -------------------------------------------------------------------------------- /predict_trimap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/predict_trimap.py -------------------------------------------------------------------------------- /test/1803151818-00000003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/test/1803151818-00000003.jpg -------------------------------------------------------------------------------- /test/1803151818-00000003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/test/1803151818-00000003.png -------------------------------------------------------------------------------- /test/1803151818-00000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/test/1803151818-00000004.jpg -------------------------------------------------------------------------------- /test/1803151818-00000004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/test/1803151818-00000004.png -------------------------------------------------------------------------------- /test/1803250719-00000103.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/test/1803250719-00000103.jpg -------------------------------------------------------------------------------- /test/1803250719-00000103.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/test/1803250719-00000103.png -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shirhe-Lyh/deep_image_matting_pytorch/HEAD/train.py --------------------------------------------------------------------------------