tensorflow / models

Models and examples built with TensorFlow
Other
77.16k stars 45.75k forks source link

What is the default Backbone for the Android Based Segmentation Model? #9985

Closed mufaawan closed 3 years ago

mufaawan commented 3 years ago

Please answer the following question for yourself before submitting an issue.

1. The entire URL of the documentation with the issue

https://www.tensorflow.org/lite/examples/segmentation/overview

2. Describe the issue

I want to train Segmentation Model on my custom dataset. I am not sure with which network backbone should I train the model. If I get to know the Train Parameters of the default segmentation model added here(https://github.com/tensorflow/examples/tree/master/lite/examples/image_segmentation/android). I am sure that would be enough help for me.

srjoglekar246 commented 3 years ago

The model has been converted from the DeepLabv3 model that was mentioned in this blog post. I assume the model must be same as this TFHub one.