一些更快的 rcnn 模型在 TF Detect 应用程序上崩溃。答案

【问题标题】：Crash on TF Detect app for some faster rcnn models.一些更快的 rcnn 模型在 TF Detect 应用程序上崩溃。
【发布时间】：2017-11-28 13:09:27
【问题描述】：

我正在训练一些更快的 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: 在 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: 在 org.tensorflow.contrib.android.TensorFlowInferenceInterface.fetch(TensorFlowInferenceInterface.java:432) 11-28 10:56:27.002 15666 16593 E AndroidRuntime: at org.tensorflow.demo.TensorFlowObjectDetectionAPIModel.recognizeImage(TensorFlowObjectDetectionAPIModel.java:167)

在 TensorFlowObjectDetectionAPIModel 类的以下代码行：

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

我试过这两个模型：

faster_rcnn_resnet101_pets fast_rcnn_inception_v2_pets

您知道导致此异常的原因吗？

问候，

【问题讨论】：

标签： tensorflow object-detection-api

【解决方案1】：

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

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

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

【讨论】：

谢谢，这也是我的问题的解决方案。在他们的代码中，max_results 的值设置为 100。当我尝试减小它时，我仍然遇到相同的错误，但是当我将其设置为 500 时，没有引发异常溢出，应用程序正在运行。谢谢！