深層学習を用いて画像判定AIを作ってみよう。(その4画像判定・改善)

今回は、画像判定のプログラムを作っていきます。

以前に、”tiger”,”snake”,”monkey”の3つの画像を保存しましたが、今回作る画像判定は、ある画像を渡したときに、この3つのうちどれに該当するかというものです。

以下がどの画像かを推定するためのプログラムです。

 

 

このプログラムの引数に画像判定したいファイルをおき実行します。

今回は、画像収集に使用した”tiger”の画像の1つを名前を変更し使用しました。この判定に使う画像は、predict.pyと同じディレクトリに置いておきます。

python predict.py tiger.jpg“と実行した結果が以下のようになります。

 

 

実行結果で、tiger(100%)と出力されていますね。これは訓練で使用している画像データなのでちゃんと画像判定ができていますが、使用する画像に、その動物以外のものや、背景が大きく写っているものは、画像判定がきちんとできない場合があります。

これを改善していくには、画像の中の領域を検知したり、そもそも訓練用の画像データに適切でないものを取り除いたりと改善の余地は多くあります。

まず画像のデータ量が少ないと思われますが、質の良い画像はそんなに多く転がっていないので、その画像を回転させて使いまわしていきます。画像を少し回転させた画像は別データとして扱われるので、1枚の画像を±20~0度の範囲で、1枚の画像を5枚分としたものと左右反転させたものを加えて増量していきます。以下が改善したプログラムです。

 

 

 

 

これで、新しいNPYデータを保存するので、”image_data_aug.npy“として名前を変更しています。また、”image_cnn.py“の中で、”image_data.npy“となっている部分を”image_data_aug.npy“に変更します。また同じように、”image_cnn.h5“となっている部分を”image_cnn.h5″と変更します。

これで、”python image_cnn.py”と実行したところ、以下のようになりました。訓練用データ、テスト用データは微量ですがaccuracyは増加しました。

 

 

他にも先ほど述べたように改善する余地としては、

  • 画像の選定
  • image_cnn.pyで”model.fit(X, y, batch_size=32, epochs=100)”ここのbatch_sizeやepochsの値を大きくする。

実際に、上の画像が、epochs=200としたもの、下の画像が、batch_size=128としたものです。どちらもテスト用データのAccuracyが増加しています。

 

 

 

画像判定のプログラムの紹介は以上になります。

 

 

コメント