如何使用 tensorflow lite 在分类对象周围绘制边界框？

Question

我想知道是否可以使用 Tensorflow lite 绘制边界框。我已经能够在 1.12 版中使用 tensorflow-android 绘制它们，但我没有在 tensorflow lite 中绘制边界框的示例。

在下面的代码中，您可以看到我在 tensorflow-android 1.12 中获取正在工作的 outputLocations 的方式。

 inferenceInterface.run(outputNames, logStats);
LOGGER.d("End Section run " + System.currentTimeMillis());
Trace.endSection();

// Copy the output Tensor back into the output array.
Trace.beginSection("fetch");
LOGGER.d("Begin Section fetch " + System.currentTimeMillis());
outputLocations = new float[MAX_RESULTS * 4];
outputScores = new float[MAX_RESULTS];
outputClasses = new float[MAX_RESULTS];
outputNumDetections = new float[1];
inferenceInterface.fetch(outputNames[0], outputLocations);

如果您能告诉我如何使用 runInference() 从 trensorflow-lite 获取 outputLocations，那就太好了。

Answer 1

如果您使用以下对象检测模型：http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18.tar.gz 输出张量已经有输出位置、分数、类别等。 您可以按照类似于 Android Java 示例应用程序的示例进行操作： https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/examples/android/app/src/main/java/org/tensorflow/demo/TFLiteObjectDetectionAPIModel.java