Hub: Why is there no pretrained object detection models trained on COCO on TF Hub?

[zacwellmer]

A starting point could just be porting the models listed in tf model zoo

[gowthamkpr]

@zacwellmer We do have a few modules in tensorflow hub already like
https://tfhub.dev/google/tf2-preview/mobilenet_v2/classification/4
https://tfhub.dev/google/imagenet/resnet_v2_152/feature_vector/4 etc..,

Please go through these modules. Also if I have misinterpreted your question please correct me. Thanks!

[zacwellmer]

Both of those links are trained on the imagenet dataset and not the coco dataset

[gowthamkpr]

@zacwellmer You can train them on coco dataset by using hub.KerasLayer(url, trainable=True) as you can finetune the model for the dataset you want.

[gowthamkpr]

@zacwellmer Can I close this issue?

[zacwellmer]

well I don't think either of your suggestions actually apply for two reasons:
1) Both resnet and mobilenet are for classification not object detection(COCO is object detection)
2) Isn't the point of TF Hub to offer pretrained models? Seems like COCO is a pretty great dataset to offer a pretrain a model on

[gowthamkpr]

@zacwellmer We do have pretrained modules on object detection here. If you can mention a specific model architecture that is not in hub which will help the community, we can take a look into it.

[gowthamkpr]

@zacwellmer Can you please respond to the above comment. Thanks!

[zacwellmer]

Maybe I'm misunderstanding the purpose of Tensorflow Hub. I was under the impression that the purpose of Tensorflow Hub is hosting models trained from different architectures or datasets for folks who don't have the compute to train the model themself.

If my understanding above is correct, that means hosting a version trained on COCO would be useful. Otherwise, the focus is on giving users a model for a specific architecture and leaving it up to them to find the compute to train on a large dataset like COCO

[vbardiovskyg]

Hi Zac, thanks for the interest! And you are right that that is the whole purpose of TensorFlow Hub and transfer learning in general.

I will relabel this as a module request to keep track.

[ChanchalKumarMaji]

Can I port some models from - here.

[ManishAradwad]

I would like to help porting these models. I tried finding some documentation about it but couldn't find anything. Can anyone plz help me on how to do it?

[j-grimstad]

Hi @ChanchalKumarMaji and @ManishAradwad, we'd be happy accept external contributions! See our documentation here: https://github.com/tensorflow/hub/tree/master/tfhub_dev

Please let us know if you have any questions.