如何在iOS中使用Tensorflow訓練過的機器學習模型

Question

我想要識別使用Tensorflow訓練過的圖像。我跟着這個steps 最後我成功地訓練了我自己的數據集並給出了一個很好的預測結果。所有的代碼都在python中。現在我試圖在iOS項目中使用這個訓練好的模型。我正在關注tutorial以在我的iOS項目中使用我訓練有素的模型。但是當我遵循這些步驟時，我在我的mac終端窗口中出現錯誤，如 -

"ERROR: /tensorflow/tensorflow/core/kernels/BUILD:2235:1: C++ compilation of rule '//tensorflow/core/kernels:self_adjoint_eig_v2_op' failed: gcc failed: error executing command /usr/bin/gcc -U_FORTIFY_SOURCE -fstack-protector -Wall -B/usr/bin -B/usr/bin -Wunused-but-set-parameter -Wno-free-nonheap-object -fno-omit-frame-pointer -g0 -O2 '-D_FORTIFY_SOURCE=1' -DNDEBUG ... (remaining 124 argument(s) skipped): com.google.devtools.build.lib.shell.BadExitStatusException: Process exited with status 4. gcc: internal compiler error: Killed (program cc1plus)"。

這個問題的解決方案是什麼？或者有什麼辦法將張量流模型轉換爲iOS支持的Core ML Model？在這裏，我分享了該錯誤的屏幕截圖。請幫助我。謝謝。

Answer 1

這很可能是由於內存不足泊塢窗容器內的巴澤爾與默認參數運行。嘗試重新運行（在bazel clean之後）具有這些額外標誌的構建到構建命令：--local_resources 2048,2.0,1.0 -j 1。您也可以嘗試給多一點資源：--local_resources 4096,2.0,1.0 -j 1

下面是此問題的一些鏈接：https://github.com/tensorflow/tensorflow/issues/1530 Tensorflow Serving Compile Error Using Docker on OSX