LoginSignup
2
9

More than 5 years have passed since last update.

chainerでサクッとアイドル画像分類

Last updated at Posted at 2017-06-10

勉強がてらトミーさんのブログを見ながら画像分類を試しました。
最適化関数とファインチューニングで一番精度のいいのを探しました。

分類する画像は秋元真夏、白石麻衣、西野七瀬、生田絵梨花、の画像です。
まずは名前の勉強から。顔と名前が一致するようにする。

ソース
https://github.com/miyamotok0105/imagenet_nogizaka

秋元真夏

スクリーンショット 2017-06-09 19.41.41.png

白石麻衣

mig.jpeg

西野七瀬

スクリーンショット 2017-06-09 19.44.50.png

生田絵梨花

スクリーンショット 2017-06-09 19.47.15.png

データセットを取ってくる

https://www.dropbox.com/sh/3i3r68f63nzis9k/AAAeAI1W3FeyXWZWGe2PcRTQa?dl=0
参考:http://toxweblog.toxbe.com/2016/10/01/chainer-%E3%83%A1%E3%83%A2%EF%BC%88%E3%81%9D%E3%81%AE%EF%BC%93%EF%BC%89%E6%A9%9F%E6%A2%B0%E5%AD%A6%E7%BF%92%E3%81%A7%E4%B9%83%E6%9C%A8%E5%9D%82%E3%83%A1%E3%83%B3%E3%83%90%E3%83%BC%E3%81%AE%E9%A1%94/

コーディングについて

これまでの分類のコードを使ってサクッと学習して推定。

chainerのexampleに推定処理を追加する「imagenet編」
http://qiita.com/miyamotok0105/items/d7cbf32667d831153cd2
chainerのexampleをcaffeモデルでファインチューニング「imagenet編」
http://qiita.com/miyamotok0105/private/5e31dc2d960dba71f85c

学習結果

参考までにグラフを添付。

normal or alexnet fine tunning

スクリーンショット 2017-06-09 21.59.52.png

normal or alexnet fine tunning 平均画像引いたバージョン

スクリーンショット 2017-06-09 22.05.19.png

normal or alexnet fine tunning:SGD

スクリーンショット 2017-06-09 22.09.23.png

normal or alexnet fine tunning:MomentumSGD

スクリーンショット 2017-06-09 22.11.50.png

normal or alexnet fine tunning:AdaGrad

スクリーンショット 2017-06-09 22.15.17.png

normal or alexnet fine tunning:RMSprop

スクリーンショット 2017-06-09 22.19.04.png

最適化アルゴリズム

validation accuracyで8割程うまくいってるアルゴリズムが何個かあります。

スクリーンショット 2017-06-09 22.29.49.png
スクリーンショット 2017-06-09 22.29.57.png

精度が出ない場合

今回は8割の精度が出ているが、全く精度が出ない場合もある。
例えば顔の感情分類は試してみていただけるとわかるのですが、精度が出にくいです。

普通に学習してもダメなのでLBPで画像を加工してから学習すると5割を超えたという報告。
色々やっても5割の精度なんです。
http://qiita.com/miyamotok0105/private/8d348996b38c9fe5c9f9

実際やってみた結果
スクリーンショット 2017-06-10 13.06.51.png

スクリーンショット 2017-06-10 13.06.58.png

学習する画像をエッジ検出後データに変更したが全然良くない

loss.png
accuracy.png

参考

cnn
http://toxweblog.toxbe.com/2016/10/01/chainer-%E3%83%A1%E3%83%A2%EF%BC%88%E3%81%9D%E3%81%AE%EF%BC%93%EF%BC%89%E6%A9%9F%E6%A2%B0%E5%AD%A6%E7%BF%92%E3%81%A7%E4%B9%83%E6%9C%A8%E5%9D%82%E3%83%A1%E3%83%B3%E3%83%90%E3%83%BC%E3%81%AE%E9%A1%94/
fine-tuning
http://toxweblog.toxbe.com/2016/12/22/chainer%E3%81%A7fine-tuning/
http://seiya-kumada.blogspot.jp/2016/02/chainer.html
network
http://www.topbots.com/a-brief-history-of-neural-network-architectures/?utm_medium=article&utm_source=facebook&utm_campaign=history-nn-architecture&utm_content=groups

2
9
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
2
9