五郎島金時の形５種類をAIで画像判定するというチャレンジはまだ道半ば

2018.10.072023.02.23

この記事は約4分で読めます。

この夏から全景AIセミナーに通っている。その学習成果として五郎島金時（加賀野菜のさつまいも）の形を判定するAIができないかどうかにチャレンジしている。セミナーで活用しているGoogleコラボのノートブックを流用して、パイトーチ、パイソンを使い試行錯誤してみた。
なれない専門用語に四苦八苦しながら、ようやく最終的な判定画像の表示ができるまでになった。しかし、正解率がぜんぜんだめ。経過をよく見るとまだまだ途中でいくつかのエラーがあり、修正が必要なようだ。

このページの目次

五郎島金時の形５種類をAIで画像判定するというチャレンジ

五郎島金時の形５種類をAIで画像判定するというチャレンジ

パイソンを使い試行錯誤

序盤で悩んだのはデータセットの準備で、ファイルリストとラベルリストを作成することだった。

zenkei_ai パッケージで提供する data_generator は、以下の２種類のリストを準備すれば使える
画像のファイルのリスト
その画像の分類されたクラス（整数）のリスト
犬猫分類については２種類の分類なのでクラスは 0 か 1 となるが、五郎島金時では５つのクラスなので、0,1,2,3,4となる。
５つのクラスは以下のように設定する。
shuu-mijikate（秀短手）：４
shuu-nagate（秀長手）：３
marushuu-mijikate（○秀短手）：２
marushuu-nagate（○秀長手）：１
ryou（良）：0

５つのクラスに分類するのだが、必要なデータセットのファイルは２つである。
画像ファイルは1018あるので、それに対応したクラスのファイルも1018必要だということだ。

ようやくうまくいったのがこれ。

import glob
gorojima_data_files = glob.glob(‘Data/gorojima/*/*.JPG’)

gorojima_data_labels = [] for f in gorojima_data_files:
c = f.split(‘/’)[-2] if c == ‘shuu-mijikate’:
imo = 4
elif c == ‘shuu-nagate’:
imo = 3
elif c == ‘marushuu-mijikate’:
imo = 2
elif c == ‘marushuu-nagate’:
imo = 1
elif c == ‘ryou’:
imo = 0
gorojima_data_labels.append(imo)

len(gorojima_data_files), len(gorojima_data_labels)

gorojima_data_filesの中身は
Data/gorojima/marushuu-mijikate/IMG_2444.JPG
Data/gorojima/shuu-mijikate/IMG_3589.JPG
…
というような画像のファイル名が1018並んでいるファイルである。

gorojima_data_filesの中身は
2
4
…
というように、0から4までのクラス名が並んでいるファイルである。

gorojima_data_filesの画像確認を行う。