首页 > 解决方案 > 如何在 android studio 中使用 tflite 进行推理

问题描述

我是机器学习的新手。我想使用east-text-detection 来识别图像中的一系列字符。我使用的模型是从 tensorflow hub 下载的。https://tfhub.dev/sayakpaul/lite-model/east-text-detector/fp16/1

我在 java 中创建了一个项目,并通过解释器加载了模型。

try {
        tflite = new Interpreter(loadModelFile());
    }catch (Exception ex){
        ex.printStackTrace();
    }

我不明白如何做出推断。首先我需要一个输入,它应该是包含要识别的句子的图像。但看看模型的特性,它接受的输入是浮点数 32,尺寸为 320x320。 模型描述 如何将图像“转换”为输入格式?然后如何打印从识别中获得的结果?

几天来我一直在发疯,理论上这应该是一件简单的事情。

标签: androidinference

解决方案


推荐阅读