Как нарисовать ограничивающие рамки вокруг классифицированных объектов с помощью tenorflow lite? - PullRequest
Новая
       6

Как нарисовать ограничивающие рамки вокруг классифицированных объектов с помощью tenorflow lite?

0 голосов
/ 22 января 2019

Я хотел бы знать, можно ли нарисовать ограничивающие рамки с помощью Tensorflow lite.Я смог нарисовать их с помощью tenorflow-android в версии 1.12, но у меня нет примера для рисования ограничивающих рамок в tenorflow lite.

В приведенном ниже коде вы можете увидеть мой путь в tenorflow-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);

Было бы замечательно, если бы вы могли сказать мне, как получить outputLocationsвместо этого используя runInference () из trensorflow-lite.

1 Ответ

0 голосов
/ 08 февраля 2019

Если вы используете модели обнаружения объектов, такие как: 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

...