CODEWITHSUNDEEP
tensorflowtensorflow-lite
tleyden
tleyden

Reputation: 1980

How was the ssd_mobilenet_v1 tflite model in TFHub trained?

How do I find more info on how the ssd_mobilenet_v1 tflite model on TFHub was trained?

Was it trained in such a way that made it easy to convert it to tflite by avoiding certain ops not supported by tflite? Or was it trained normally, and then converted using the tflite converter with TF Select and the tips on this github issue?

Also, does anyone know if there's an equivalent mobilenet tflite model trained on OpenImagesV6? If not, what's the best starting point for training one?

Upvotes: 1

Views: 487

Answers (1)

Sachin Joglekar
Sachin Joglekar

Reputation: 696

I am not sure about about the exact origin of the model, but looks like it does have TFLite-compatible ops. From my experience, the best place to start for TFLite-compatible SSD models is with the TF2 Detection Zoo. You can convert any of the SSD models using these instructions.

To train your own model, you can follow these instructions that leverage Google Cloud.

Upvotes: 1

Related Questions