search

anilsathyan7 / Portrait-Segmentation

Real-time portrait segmentation for mobile devices
MIT License
638 stars 133 forks source link

Mask overlay calculator CPU #41

Open Nerdyvedi opened 3 years ago

Nerdyvedi commented 3 years ago

Hi, I am trying to run the model on CPU, I wrote a custom calculator, which simply multiplies input with mask and background with bitwise_not of mask, But the output I am getting is not good.

But, when I build the model for Android and use the mask overlay calculator, The result is much better. Help would mean a lot. Thanks

anilsathyan7 commented 3 years ago

Mask overlay calculator will have feathering around the edges by default, so it would look nicer. Also, i beleive it runs on GPU so it should run faster. Otherwise both should produce similar results, as long as the internlal logic is same.