ピカチュウとミミックの画像をAIで判別するモデルの作り方

今回ご紹介する手順はMacが必要となりますが、パソコンの知識がない方でもわかりやすい様に解説します。

AIは何なのかよくわかっていない方も必ずAIという単語は耳にしたことがあると思います。

AIは簡単にいうと、大量のデータを元に学習をし「分類」や「回帰」を行うものです。

今回はこの「分類」にフォーカスして説明を行います。

画像を使った分類

ポケモンの「ピカチュウ」と「ミミッキュ」ってよく似ていますよね？

初めてみたときは進化前か進化後がわからないけど同じものとして認識していました。

そこで、「ピカチュウ」と「ミミッキュ」の画像を２０枚ずつかき集めて学習させます。

手順１

「学習用データ」というフォルダを作成しその中に「ピカチュウ」と「ミミッキュ」フォルダを作ります。

手順２

Googleの画像検索を使い「ピカチュウ」と「ミミッキュ」の画像をフォルダにドラッグアンドドロップし画像をかき集めます。（２０枚程度）

手順３

同様に「テスト用データ」フォルダとその中に「ピカチュウ」と「ミミッキュ」フォルダを作成し、手順２以外の画像を入れます。（５枚程度）

手順４

XcodeのPlaygroundを開き下記の３行を書いてください。

import CreateMLUI

let builder = MLImageClassifierBuilder()
builder.showInLiveView()

なんとプログラムソースはこの３行だけでOKなんです。

importでエラーになる方はこちらを参考にしてください。

CreateMLUIをimportすると「No such module ‘CreateMLUI’」のエラーになる

手順６

画像データを学習させます。

プログラムを実行すると下記の画面が表示されますので「学習用データ」フォルダごとドラッグアンドドロップしてください。

実行するとこんな感じになります。

手順７

同様に、「テスト用データ」フォルダをドラッグアンドドロップします。

結果

このように、ピカチュウとミミッキュの画像それぞれが正しくラベルづけされています。

画面左下の「Predicted」はAIが出した答え右下の「True」が実際の答えです。

黄色いピカチュウ、白いピカチュウ、イラスト、ぬいぐるみを織り交ぜましたが

今回は１００％の確率で判定ができました。

AIはすばらしいですね。

これを応用するとどの芸能人に似ているかのアプリも作れちゃうんです。