Search code examples
androidpythontensorflowobject-detectionfaster-rcnn

How to make tensorflow object detection faster-r cnn model work on Android?


I have a question about Tensorflows Object Detection API. I trained Faster R-CNN Inception v2 model with my own dataset for Traffic Sign Classification and I want to deploy it to Android but Tensorflows Object Detection API for Android and/or Tensorflow Lite seems only supporting SSD models.

Is there any way to deploy a Faster R-CNN model to Android? I mean how can I put my frozen inference graph of Faster R-CNN to android API instead of SSDs frozen inference graph ?


Solution

  • For SSD models, it must be possible using the export_tflite_ssd_graph.py tool. But:

    At this time only SSD models are supported. Models like faster_rcnn are not supported at this time

    See this guide for more information.