如何从tflite模型输出中提取边界框信息

时间:2020-11-12 02:47:11

标签: java android tensorflow tensorflow-lite

enter image description here

如您在图像中看到的,这是我的tflite模型输出。

我有用于从中提取信息的代码

 outputLocations = new float[1][NUM_DETECTIONS][4];
        outputClasses = new float[1][NUM_DETECTIONS];
        outputScores = new float[1][NUM_DETECTIONS];
        numDetections = new float[1];

        Object[] inputArray = {imgData};
        Map<Integer, Object> outputMap = new HashMap<>();
        outputMap.put(0, outputLocations);
        outputMap.put(1, outputClasses);
        outputMap.put(2, outputScores);
        outputMap.put(3, numDetections);

但是我不明白为什么会这样吗?出现此顺序的任何特殊原因。

0 个答案:

没有答案