TF Detect应用程序崩溃了一些更快的rcnn模型。

时间:2017-11-28 13:09:27

标签: tensorflow object-detection-api

我正在训练一些只有一个课程的更快的rcnn模型 我自己的数据集。培训顺利,但出口时 推理模型,并尝试在Android示例应用程序上使用它 (TF检测),我得到以下异常:

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:at java.nio.HeapFloatBuffer.put(HeapFloatBuffer.java:179) 11-28 10:56:27.002 15666 16593 E AndroidRuntime:at org.tensorflow.Tensor.writeTo(Tensor.java:488) 11-28 10:56:27.002 15666 16593 E AndroidRuntime:at 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日:56:27.002 15666 16593ËAndroidRuntime:在org.tensorflow.demo.TensorFlowObjectDetectionAPIModel.recognizeImage(TensorFlowObjectDetectionAPIModel.java:167)

来自TensorFlowObjectDetectionAPIModel类的以下代码行:

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

我试过这两个模型:

faster_rcnn_resnet101_pets faster_rcnn_inception_v2_pets

您是否知道导致此异常的原因是什么?

此致

1 个答案:

答案 0 :(得分:1)

我正在使用FasterRcnn,我通过在TensorFlowObjectDetectionAPIModel.java(TensorFlow android示例应用程序)中将MAX_RESULTS的值从100更改为500来解决同样的问题

问题是RCNN返回的输出多于SSD

对不起,这是我的第一个回答