├── README.md ├── config.py ├── dataloader.py ├── evaluate.py ├── imgs ├── result │ ├── PrismaMattingNet_test1_alpha.jpg │ ├── PrismaMattingNet_test1_blend.jpg │ ├── PrismaMattingNet_test2_alpha.jpg │ ├── PrismaMattingNet_test2_blend.jpg │ ├── PrismaMattingNet_test3_alpha.jpg │ ├── PrismaMattingNet_test3_blend.jpg │ ├── PrismaMattingNet_test4_alpha.jpg │ ├── PrismaMattingNet_test4_blend.jpg │ ├── PrismaMattingNet_test5_alpha.jpg │ ├── PrismaMattingNet_test5_blend.jpg │ ├── PrismaMattingNet_test6_alpha.jpg │ ├── PrismaMattingNet_test6_blend.jpg │ ├── PrismaMattingNet_test7_alpha.jpg │ ├── PrismaMattingNet_test7_blend.jpg │ ├── PrismaMattingNet_test8_alpha.jpg │ ├── PrismaMattingNet_test8_blend.jpg │ ├── PrismaNet_test1_alpha.jpg │ ├── PrismaNet_test1_blend.jpg │ ├── PrismaNet_test2_alpha.jpg │ ├── PrismaNet_test2_blend.jpg │ ├── PrismaNet_test3_alpha.jpg │ ├── PrismaNet_test3_blend.jpg │ ├── PrismaNet_test4_alpha.jpg │ ├── PrismaNet_test4_blend.jpg │ ├── PrismaNet_test5_alpha.jpg │ ├── PrismaNet_test5_blend.jpg │ ├── PrismaNet_test6_alpha.jpg │ ├── PrismaNet_test6_blend.jpg │ ├── PrismaNet_test7_alpha.jpg │ ├── PrismaNet_test7_blend.jpg │ ├── PrismaNet_test8_alpha.jpg │ ├── PrismaNet_test8_blend.jpg │ ├── SegNet_test1_alpha.jpg │ ├── SegNet_test1_blend.jpg │ ├── SegNet_test2_alpha.jpg │ ├── SegNet_test2_blend.jpg │ ├── SegNet_test3_alpha.jpg │ ├── SegNet_test3_blend.jpg │ ├── SegNet_test4_alpha.jpg │ ├── SegNet_test4_blend.jpg │ ├── SegNet_test5_alpha.jpg │ ├── SegNet_test5_blend.jpg │ ├── SegNet_test6_alpha.jpg │ ├── SegNet_test6_blend.jpg │ ├── SegNet_test7_alpha.jpg │ ├── SegNet_test7_blend.jpg │ ├── SegNet_test8_alpha.jpg │ └── SegNet_test8_blend.jpg └── test │ ├── test1.jpg │ ├── test2.jpg │ ├── test3.jpg │ ├── test4.jpg │ ├── test5.jpg │ ├── test6.jpg │ ├── test7.jpg │ └── test8.jpg ├── legacy ├── PrismaNet_Sigmoid.py └── train_prisma_sigmoid.py ├── loss.py ├── main.py ├── models ├── PrismaMattingNet_portrait_epoch-0099.pth ├── PrismaNet_portrait_epoch-0099.pth └── SegNet_portrait_epoch-0000.pth ├── nets ├── PrismaMattingNet.py ├── PrismaNet.py └── segnet.py ├── onnx2coreml.py ├── test.py ├── train_prisma.py ├── train_prisma_matting.py └── train_segnet.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/config.py -------------------------------------------------------------------------------- /dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/dataloader.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/evaluate.py -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test1_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test1_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test1_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test1_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test2_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test2_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test2_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test2_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test3_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test3_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test3_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test3_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test4_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test4_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test4_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test4_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test5_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test5_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test5_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test5_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test6_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test6_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test6_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test6_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test7_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test7_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test7_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test7_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test8_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test8_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaMattingNet_test8_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaMattingNet_test8_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test1_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test1_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test1_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test1_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test2_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test2_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test2_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test2_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test3_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test3_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test3_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test3_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test4_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test4_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test4_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test4_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test5_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test5_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test5_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test5_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test6_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test6_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test6_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test6_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test7_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test7_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test7_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test7_blend.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test8_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test8_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/PrismaNet_test8_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/PrismaNet_test8_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test1_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test1_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test1_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test1_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test2_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test2_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test2_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test2_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test3_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test3_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test3_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test3_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test4_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test4_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test4_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test4_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test5_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test5_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test5_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test5_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test6_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test6_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test6_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test6_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test7_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test7_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test7_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test7_blend.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test8_alpha.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test8_alpha.jpg -------------------------------------------------------------------------------- /imgs/result/SegNet_test8_blend.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/result/SegNet_test8_blend.jpg -------------------------------------------------------------------------------- /imgs/test/test1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test1.jpg -------------------------------------------------------------------------------- /imgs/test/test2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test2.jpg -------------------------------------------------------------------------------- /imgs/test/test3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test3.jpg -------------------------------------------------------------------------------- /imgs/test/test4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test4.jpg -------------------------------------------------------------------------------- /imgs/test/test5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test5.jpg -------------------------------------------------------------------------------- /imgs/test/test6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test6.jpg -------------------------------------------------------------------------------- /imgs/test/test7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test7.jpg -------------------------------------------------------------------------------- /imgs/test/test8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/imgs/test/test8.jpg -------------------------------------------------------------------------------- /legacy/PrismaNet_Sigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/legacy/PrismaNet_Sigmoid.py -------------------------------------------------------------------------------- /legacy/train_prisma_sigmoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/legacy/train_prisma_sigmoid.py -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/main.py -------------------------------------------------------------------------------- /models/PrismaMattingNet_portrait_epoch-0099.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/models/PrismaMattingNet_portrait_epoch-0099.pth -------------------------------------------------------------------------------- /models/PrismaNet_portrait_epoch-0099.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/models/PrismaNet_portrait_epoch-0099.pth -------------------------------------------------------------------------------- /models/SegNet_portrait_epoch-0000.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/models/SegNet_portrait_epoch-0000.pth -------------------------------------------------------------------------------- /nets/PrismaMattingNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/nets/PrismaMattingNet.py -------------------------------------------------------------------------------- /nets/PrismaNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/nets/PrismaNet.py -------------------------------------------------------------------------------- /nets/segnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/nets/segnet.py -------------------------------------------------------------------------------- /onnx2coreml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/onnx2coreml.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/test.py -------------------------------------------------------------------------------- /train_prisma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/train_prisma.py -------------------------------------------------------------------------------- /train_prisma_matting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/train_prisma_matting.py -------------------------------------------------------------------------------- /train_segnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/digital-nomad-cheng/Mobile_Portrait_Matting_PyTorch/HEAD/train_segnet.py --------------------------------------------------------------------------------