1

我正在训练一些更快的 rcnn 模型,在我自己的数据集上只有一个类。训练进行得很好,但是当我导出模型进行推理并尝试在 android 示例应用程序(TF Detect)上使用它时,我得到了以下异常:

11-28 10:56:27.002 15666 16593 E AndroidRuntime: 致命异常: 推断 11-28 10:56:27.002 15666 16593 E AndroidRuntime: Process: org.tensorflow.demo, PID: 15666 11-28 10:56:27.002 1566 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 16593 E AndroidRuntime:在 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: 在 org.tensorflow.contrib.android.TensorFlowInferenceInterface.fetch(TensorFlowInferenceInterface.java:432) 11-28 10:56:27.002 15666 16593 E AndroidRuntime: 在 org.tensorflow.demo。TensorFlowObjectDetectionAPIModel.recognizeImage(TensorFlowObjectDetectionAPIModel.java:167)

在 TensorFlowObjectDetectionAPIModel 类的以下代码行中:

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

我尝试了这两个模型:

fast_rcnn_resnet101_pets faster_rcnn_inception_v2_pets

您对导致此异常的原因有任何想法吗?

问候,

4

1 回答 1

1

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

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

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

于 2018-06-11T16:07:16.733 回答