新潟県工業技術総合研究所は、工業系の技術支援機関です。

Topページ　>　機械・金属関係　技術トピックス 　>　Lobeによる金属組織の認識

Lobeによる金属組織の認識

１．はじめに

　先日、マイクロソフト社からLobeという機械学習のアプリが公開されました。令和2年11月13日時点でベータ版が公開されており、ダウンロードしてローカル環境で無料で使用することができます。今のところ、入力データは画像のみとなっています。
　このアプリは学習から推論までを簡単な操作でプログラム無しで実行できるという大きな特徴を持っているため、これを機に機械学習を始める方が格段に増えることが予想されます。さらに、アプリで学習した結果（モデル）は、Tensorflowなどで使えるよう出力可能となっているため、AI・IoT分野の技術者にとってもたいへん有用と考えられます。
　今回はLobeを使って鉄鋼材料の5種類の金属組織の学習を試みました。この実験は令和2年11月に行ったものです。

２．Lobeによる金属組織の認識

　機械学習では、あらかじめラベル付けされた画像を用意しておくと後処理が容易になります。ここでは、5種類（オーステナイト、セメンタイト、フェライト、マルテンサイト、パーライト）の金属組織の画像^1）を図1に示すようにフォルダに格納しておきます。各フォルダにはサイズ1280×960の画像を20枚ずつ格納しました。このアプリではインポートする画像のサイズに制約はないようです。
　フォルダに格納した画像をデータセットとしてLobeにインポートします。Lobeを起動して、「New Porject」、「Import」、「Dataset」、「Choose Dataset」の順にクリックしていくと、フォルダの選択ダイアログが表示されます。ここで、先ほど作成したフォルダ「train」を選択し、「Label Using Folder Name」をチェックして「Import」をクリックします。これにより、画像のインポートおよびラベル付けが行われ、図2のように読み込んだ画像が表示され自動的に学習が始まります。
　ここで、画面の左上の「Training」をクリックすると図3のように学習の進捗が表示されます。学習が終了するとTrainingと表示されていた部分が図4のようにTrainに表示が変わり、学習の結果が表示されます。図4において、全体の正解率が96％となっています。正解率をさらに上げるため、図5のように画面の左上隅をクリックしてメニューを表示させて「Optimize Model...」をクリックすると、学習モデルの最適化が行われます。最適化の結果を図6に示します。図6より、正解率が100％になっていることが分かります。なお、このアプリでは、誤って認識した画像について、ラベルの変更や削除をすることができます。


図1　データセットのフォルダ構造



図2　インポート・ラベル付け終了



図3　学習中



図4　学習終了



図5　学習の最適化



図6　最適化終了

　学習結果を用いて、学習に用いていない画像データを推論します。画面の左上の「Play」をクリックすると図7のようになります。図7において、画面のブランコをしている絵の部分に学習に用いていない画像データをドラッグアンドドロップすると、図8のように推論結果が表示されます。図8において、組織画像はマルテンサイトですが、推論結果もマルテンサイトとなっています。このアプリでは、推論結果を学習結果に追加することも簡単にできます。


図7　推論画面



図8　推論結果

終わりに

　今回は、Lobeによる金属組織の認識について紹介しました。今回用いた画像については、画像の読み込みから推論まで5分程度で行うことができました。画像を金属組織から他のものに置き換えることで、様々な画像認識が可能となりますので、ぜひ試していただきたいと思います。

参考文献

問い合わせ：新潟県工業技術総合研究所 　　　　 　　 中越技術支援センター 　　斎藤　雄治 　 　　　　 　TEL：0258-46-3700 　　FAX：0258-46-6900