ソニーNeural Network Consoleで学習させたネットワークをnnabla (Neural Network Libraries)で推論するシンプルな方法

1年ほど前、ソニー Neural Network Console と エクセルで始める ニューラルネットワークを投稿しました。

ソニー Neural Network Console と エクセルで始める ニューラルネットワーク

Neural Network Console(以下、NNC)は、プログラムを組まずとも、ニューラルネットワークの設計・学習・検証・推論できるようになっています。インストールも展開するだけなので、Windowsでニューラルネットワークを始めようと思っている方には、おすすめのソフトウェアです。いろいろ使っていくと、学習結果を使て推論だけのシステムを作りたくなることでしょう。折角、学習したネットワークをクラウドやエッジで使わないのはもったいないと考えている方も多いと思います。
今回は、NNCで学習したネットワークを用いて、同じくソニーが公開しているNeural Network libraries(通称、nnabla。以下、nnabla)を使うことで、驚くほど簡単に推論システムを作る方法をご紹介します。

1.ゴール設定

このブログでは、
* AIエンジニア1がネットワークを作り、NNCで教師データを学習させて、プログラマ2にその結果を渡す
* プログラマが、その結果を使って、推論システム3を作る

というプロセスを考えます。

2.プログラマに渡すものは、results.nnp だけ

NNCで学習させた結果は、通常、results.nnpというファイルに保存されています。AIエンジニアは、このresults.nnpだけをプログラマに渡してください。4
次に、ネットワーク図とconfigタブのExcutor欄のNetwork変数の値5を伝えてください。
最後に、入力のデータ形式(JPEGファイルとかCSVファイルとか)や出力形式(画像ファイルとか標準出力で数値とか)がどうなっているか等の情報交換をしてください。6
AIエンジニアはこれでこの業務は完了です。7

3.results.nnpをチェックする

プログラマは伝えられた各種情報とresults.nnpをチェックする以下のプログラム結果を照らして、推論ネットワーク名、入出力名8を決定してください。

例えば、このようなネットワークを「Main」(既定値)として作った場合、

Mainネットワーク

自動的に、「MainValidation」と「MainRuntime」が生成され、configのExecutor欄のNetworkに「MainRuntime」が登録されています。これが推論で使用するネットワークです。

コンフィグ

先ほどのチェックプログラムを実行すると

結果はこんな感じになりますので、「MainRuntime」のInputsとOutputsの名前をメモリます。

推論プログラムの作成

先ほどメモした、「推論ネットワーク名」、「入力名」、「出力名」を以下のように指定すれば、大体終わりです。大体といったのは、実際にxに入れるデータの読み込みや出力の仕方などを書かなければなりませんので、自分のデータに合わせて記述してください。

まとめ

デフォルトのMainネットワークだけで構築したNNCの学習済みネットワークは、nnpファイルだけで、推論プログラムが簡単9に書けることがわかっていただけたと思います。このようにNNCを使うと、AIエンジニアはネットワーク設計のみに専念でき、システムエンジニアはnnablaを使った推論プログラムが可能になり、分業化10できますね。

プログラミングはちょっと不得意だけど、分析は得意だからやりたいとかディープラーニングも使った分析がしたいとかお考えのあなた!いい環境ありますよ。まずはお問い合わせください。


  1. この場合、AIエンジニアはプログラムが組めなくても問題ありません。 
  2. ここでは、pythonプログラマを想定しています。ここでは紹介しませんが、C++でもできます。 
  3. 入力データを入れて、対応する何か(分類なら分類結果、回帰なら予測結果)を出力するシステムのことです。 
  4. ネットワーク構造ファイル(net.nntxt)やNNC上でExportできるpythonコードはいりません。 
  5. 推論ネットワーク名です。デフォルトは”MainRuntime”です。 
  6. なんなら、results.nnpが含まれているフォルダー丸ごと渡せばいいんですけどね。 
  7. 次の案件に移りましょう。 
  8. 入力が1つで、出力が1つとは限りません。 
  9. ほとんどの場合、推論ネットワーク=MainRuntime、入力名=Inputなので、出力名だけ変えればよいだけです。 
  10. 個人として分業化がいいかどうかわかりませんが、会社的な効率は上がるハズです。