トップサービス導入事例強み開発ブログよくある質問会社案内 お問い合わせ
Dev Blog

TensorflowをGo言語で使う

画像認識において、 (1)大量の画像を学習させてモデルを作る (2)その後、学習させたモデルを使って input画像を判定する

というのが大きな流れです。 (1)はPythonとKeras(Tesorflow)でゴリゴリやる (2)の判定処理もPythonで作る

でやっていたのですが、両方ともパフォーマンスが著しく悪い。 (1)は良いGPUマシンを使うしかない。 (2)は良いGPUマシンでも改善しない。マシン環境にもよりますが、6―7秒かかったりするのでさすがに厳しい。

そこで回避策を探していて行き着いたのが「Go言語」(golang)。 「Install TensorFlow for Go」 TenforflowってPython用だと思っていたのですが、違うらしい。 ・C、C++ ・Java ・Go ・Javascript でも可能。 しかし「Caution: The TensorFlow Go API is not covered by the TensorFlow API stability guarantees.」って。。 え?golangもTensorflowもGoogleじゃないの?使ってみましょう。

結果、判定のパフォーマンスが向上! 1-2秒で済むようになった。

(1)はPythonでゴリゴリやる (2)はGo言語でやる というのが良いようです。