anilsathyan7 / Portrait-Segmentation

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

Issue on Applying Custom Model on Android #52

Open haqkiemdaim opened 2 years ago

haqkiemdaim commented 2 years ago

Hi there. I would say good repo you open-sourced here! Just want to ask why the app crashes when I used another model? It works only with your deconv_fin_munet.tflite

The 'another model' that I mentioned above accepts input frame with size 224 x 224. So I change the below declaration (in ImageSegmentorFloatMobileUnet.java) to 224:

int opsize = 224 getImageSizeX() {return 224;} getImageSizeY() {return 224;}

It successfully built with few warnings. When I click the app on my device, it crash.

Your response regarding this is so much appreciated. Thank you!