適用於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異常，因此我相信在將權重加載到模型時會遇到此錯誤。