Skip to content
This repository was archived by the owner on Jun 14, 2024. It is now read-only.
This repository was archived by the owner on Jun 14, 2024. It is now read-only.

MLKit's on-device Label(Image) Detection is very poor when compared to TensorFlow Label Detection #17

@Raju-567

Description

@Raju-567

MLKit's on-device Label Detection is very poor when compared to TensorFlow Label detection. MLKit on-device model is not able to predict the images accurately .

When I debugged the both apps, I found that MLKit uses preview ImageFormat as NV21 and TensorFlow is using YUV_420_888 Image format.

ML kit android app
https://github.com/firebase/quickstart-android

TensorFlow android app
https://github.com/tensorflow/examples/tree/master/lite/examples/image_classification/android

And I even tried using tflite model from TensorFlow app as a custom model in the MLKit android app and I see the predictions results are not satisfactory (the results are same as ML Kits on device
Label Detection)

You can see the prediction results of MLKit and TensorFlow apps on water bottle and Laptop input Images.
MLKit_LabelDetection_Laptop
MLKit_LabelDetection_WaterBottle
TensorFlow_LabelDetection_Laptop
TensorFlow_LabelDetection_WaterBottle

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions