适用于Android的Tensorflow SSD300

Question

我是一名在日本学习机器学习的大学生。 我不擅长使用英语，但是我会努力用英语传达我的情况。

我现在正在尝试将对象检测模型用于android。 我使用SSD300_mobile_net进行训练，然后得到了具有模型权重的.hdf5文件。

接下来，我将此.hdf5文件转换为适合tensorflow的.pb文件。 最后，我想将这个.pb文件用于tensorflow android。（ https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/android ）

但是我在使用设备调试应用程序时，在Android Studio上出现如下错误

这是我想知道如何解决此问题的错误捕获 。

Answer 1

我感觉这是您的硬件问题。 通常，hdf5文件非常大，并且正在执行的计算甚至更大。 由于您的系统给出了OOM异常，因此我相信在将权重加载到模型时会遇到此错误。