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);

如果您能告诉我如何使用trensorflow-lite的runInference（）获取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

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

1 个答案: