tensorflow-litetensorflow-model-garden

How to use ssd_mobilenet_v2_mnasfpn_coco in android app


I downloaded ssd_mobilenet_v2_mnasfpn_coco and untar it, found there is already a tflite file called model.tflite, I want to know how to use it in android app, can I replace original detect.tflite at object_detection android demo app with this model.tflite.

If can, is there anywhere I need to modify too?


Solution

  • Yes, assuming you mean this but you need to change few things ... In TFLiteObjectDetectionAPIModel.java change the following values to

    P.S: You also need to add some missing label values in labelmap.txt which are wrongly added as ???