我正在训练一些只有一个课程的更快的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
您是否知道导致此异常的原因是什么?
此致
答案 0 :(得分:1)
我正在使用FasterRcnn,我通过在TensorFlowObjectDetectionAPIModel.java(TensorFlow android示例应用程序)中将MAX_RESULTS的值从100更改为500来解决同样的问题
问题是RCNN返回的输出多于SSD
对不起,这是我的第一个回答