How to convert SSD .pb file to tflite

125 views
Skip to first unread message

Ohara

unread,
Mar 12, 2020, 2:27:30 AM3/12/20
to Discuss
I have followed the guide in https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html till the very last step, I'm interested in deploying my custom Object Detection model into android. How would I be to do that? I've tried different methods online where i would get errors such as (Tensor 'image_tensor' has invalid shape '[None, None, None, 3]') or (Check failed: array.data_type == array.final_data_type Array). Thank you in advanced

Rebecca

unread,
Mar 12, 2020, 2:33:32 AM3/12/20
to Ohara, Discuss
You can try MobileNet v3 for training and Tensorflow Lite for
prediction.
Those work on our Android devices.

https://github.com/frotms/MobilenetV3-Tensorflow
https://www.tensorflow.org/lite

Regards.
> --
> You received this message because you are subscribed to the Google
> Groups "Discuss" group.
> To unsubscribe from this group and stop receiving emails from it, send
> an email to discuss+u...@tensorflow.org.
> To view this discussion on the web visit
> https://groups.google.com/a/tensorflow.org/d/msgid/discuss/3ac3b610-0245-45f8-8cd6-3617d3de850a%40tensorflow.org
> [1].
>
>
> Links:
> ------
> [1]
> https://groups.google.com/a/tensorflow.org/d/msgid/discuss/3ac3b610-0245-45f8-8cd6-3617d3de850a%40tensorflow.org?utm_medium=email&utm_source=footer
Reply all
Reply to author
Forward
0 new messages