テックブログエッジコンピューティングAIこんにちは。技術記事担当のHです。今回はCONEXIOBlackBearで機械学習モデルを動かす話をいたします。AIによるリアルタイム検知などの用途で活用できるエッジコンピューティングゲートウェイです。これらの用途においてはすでに活用されており、また、今後も拡大が見込まれる技術の1つとして画像認識の機械学習モデルがありますが、私自身はまだCONEXIOBlackBear上で動かしたことがありません。そこで、本記事では、CONEXIOBlackBear上で画像認識の機械学習モデルを動かし、USBカメラにて撮影した画像をリアルタイムに認識させてみたいと思います。学習データここでは、CIFAR-10というデータセットを用います。CIFAR-10は、10種類に分類した学習用5万枚、テスト用1万枚からなる合計6万枚の画像データセットです。詳細はリンク先をご覧ください。モデルニューラルネットワークを記述するためのPythonライブラリkerasにサンプルとして含まれているcifar10_cnn.pyを用います。したがって、kerasによって記述されたモデルになります。このサンプルは、https://github.com/keras-team/keras/tree/2.2.4をダウンロードして展開したものの中のexamplesというフォルダに入っています。Network、CNN)です。入力画像は32*32ピクセルのカラー画像(RGB)です。層構成は、以下のようになっています。1畳み込み層(32*32*3入力、32チャネル出力)2畳み込み層(32チャネル出力)3MaxPooling層(サイズ2*2)4畳み込み層(64チャネル出力)5畳み込み層(64チャネル出力)6MaxPooling層(サイズ2*2)7全結合層(512ユニット)8全結合層(10ユニット=分類の数)畳み込み層のフィルタサイズは3*3です。活性化関数は、最終出力以外はReLUで、最終出力は、多クラス分類なのでsoftmax関数です。MaxPooling層の後ろと全結合層の間にDropout層があります。学習は、cifar10_cnn.pyで実行可能ですので、自分でコーディングせずそのまま動かします。学習回数は50epochとしました。この回数だと精度が75%程度だとサンプルのソースコード内に注意書きがありますが、今回はサンプルの動作確認ですので、精度にこだわらず、50エポックでよしとします。実際にやってみたところ79%となり、やはり注意書きの通りといったところです。学習は、膨大な計算が必要なため、一般にエッジデバイスではなく、計算リソースの豊富な環境で行うことが一般的です。そのため、CONEXIOBlackBearではなく、手元のPCにて行いました。所要時間はおよそ1.5hでした。準備kerasのモデルを動かすためには、kerasと、今回バックエンドとして使用しているtensorflowのインストールが必要です。kerasは、後述のtensorflowに同梱されていますので、別途インストールしてはいません。tensorflowは、CONEXIOBlackBearで動かすには、CONEXIOBlackBear用にビルドしたバージョンが必要ですので、当社でビルドしたものを使用してインストールします。※なお、当社でビルドしたtensorflowは、CONEXIOBlackBearのユーザー様へご提供可能ですので、お問い合わせください。コーディングCONEXIOBlackBearで実行するコードは以下です。約5秒ごとにカメラで撮影した画像をCNNモデルに入力して推論を行います。画像のサイズは、モデルの入力サイズに合わせてリサイズしています。学習モデルのデータは、1.3で出力されたkeras_cifar10_trained_model.h5というファイルを用いるので、PCからCONEXIOBlackBear上に配置しておきます。実際の景色を撮影しながら動作させることが難しかったので、PCの画面に表示した写真をWebカメラで撮影しました。精度は学習回数の関係上それほど高くないですが、以下の例ではautomobileが正しく判別されています。推論1回の所要時間は、50~60msecでした。今回のモデルの規模であれば、推論は十分短い時間で実行できました。このように学習済みモデルをCONEXIOBlackBearにデプロイすれば、画像認識のアプリケーションが十分実現できそうです。アプリケーション次第ですが、もっと複雑なモデルでも十分リアルタイムに実行できそうです。)モデルへの入力画像(PC画面に画像を映し、USBカメラで撮って32*32リサイズ)※サイズが小さいのでぼやけて見える。まとめ以上のように、CONEXIOBlackBearで機械学習モデルを動かせました。学習済みモデルであれば十分短時間で実行可能ということがわかりました。以前、greengrassでアプリケーションをデプロイという記事を書きました。今回のような機械学習モデルを豊富なリソースがあるクラウド側で開発して、同様の方法でエッジ側にデプロイすれば、まさに本格的なエッジコンピューティングになると思います。今後の記事で、トライしてみたいと思います。