我正在训练一些更快的 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
您对导致此异常的原因有任何想法吗?
问候,