Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Image Segmentation Android sample to support multiple models #26

Open
khanhlvg opened this issue Sep 14, 2020 · 2 comments
Open

Image Segmentation Android sample to support multiple models #26

khanhlvg opened this issue Sep 14, 2020 · 2 comments

Comments

@khanhlvg
Copy link

Currently the TFLite image segmentation sample only supports 1 segmentation model trained on COCO dataset with mobilenet_v2_0.5 backbone.

However, there are many different versions of image segmentation with different backbone architectures and trained on different datasets. We'd like to integrate these models into the sample app and allow users to try different models by selecting them from the sample app UI.

Please directly contribute pull request to the tensorflow/examples with this enhancement.

@margaretmz
Copy link
Contributor

margaretmz commented Sep 14, 2020

Just to add a bit more details to the requirements - you can refer to these 2 examples:

  • Image Classifier (Java, Camera2)
  • Cartoonizer (Kotlin, CameraX)
    • Most of the code is in CameraFragment.kt with modelSpinner. The project uses ML Model Binding so the classes for loading the models and creating interpreters etc are the generated classes under app/build/generated/ml_source_out/debug/[package-name]/ml/

@khanhlvg
Copy link
Author

FYI the TFLite team is working on updating the Android image segmentation model to demonstrate 2 ways of running segmentation models: with Task API and with Support Library + vanilla TFLite interpreter.

We should wait a few weeks before starting to work on this.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants