search

adumrewal / imageMatting

Simplified Deep Image Matting training code with keras on tensorflow
https://arxiv.org/abs/1703.03872
39 stars 6 forks source link

Mobile Matting? #2

Closed InternetMaster1 closed 4 years ago

InternetMaster1 commented 4 years ago

Is it possible to change the backbone to mobilenetv2 in this library to reduce the size and make it run on mobile devices?

Similar to this library : https://github.com/foamliu/Mobile-Image-Matting

adumrewal commented 4 years ago

Yes, that should be possible. I haven't tested it though. You can just load the model you want into the final variable and should be good to go.

Something like this: final = DeepLab(backbone='mobilenet', output_stride=16, num_classes=1)