Сбой в приложении TF Detect для некоторых более быстрых моделей rcnn.

Question

Я тренирую некоторые более быстрые модели rcnn только с одним классом на своем собственном наборе данных. Обучение идет хорошо, но когда я экспортирую модель для вывода и пытаюсь использовать ее в образце приложения для Android (TF Detect), я получаю следующее исключение:

11-28 10:56:27.002 15666 16593 E AndroidRuntime: ИСКЛЮЧИТЕЛЬНОЕ ИСКЛЮЧЕНИЕ: вывод 11-28 10:56:27.002 15666 16593 E AndroidRuntime: Процесс: org.tensorflow.demo, PID: 15666 11-28 10:56:27.002 15666 16593 E AndroidRuntime: java.nio.BufferOverflowException 11-28 10:56:27.002 15666 16593 E AndroidRuntime: в java.nio.HeapFloatBuffer.put(HeapFloatBuffer.java:179) 11-28 10:56:27,002 15666 1659 E: E в org.tensorflow.Tensor.writeTo(Tensor.java:488) 11-28 10:56:27.002 15666 16593 E AndroidRuntime: в org.tensorflow.contrib.android.TensorFlowInferenceInterface.fetch(TensorFlowInferenceInterface.java:478) 11-28 10:56:27.002 15666 16593 E AndroidRuntime: at org.tensorflow.contrib.android.TensorFlowInferenceInterface.fetch(TensorFlowInferenceInterface.java:432) 11-28 10:56:27.002 15666 16593 E AndroidRuntime: в org.tensorde.recognizeImage (TensorFlowObjectDetectionAPIModel.java:167)

в следующей строке кода из класса TensorFlowObjectDetectionAPIModel:

inferenceInterface.fetch (outputNames [0], outputLocations);

Я попробовал эти две модели:

fast_rcnn_resnet101_pets fast_rcnn_inception_v2_pets

Есть ли у вас идеи о том, что вызывает это исключение?

С Уважением,

Answer 1

Я использую FasterRcnn, я решил ту же проблему, изменив значение MAX_RESULTS со 100 на 500 в TensorFlowObjectDetectionAPIModel.java (пример приложения Android TensorFlow)

Проблема в том, что RCNN возвращает больше вывода, чем SSD

Извините, это мой первый ответ