顔画像のデータセット

ディープラーニングのモデルを顔画像で学習をしようと考え、顔画像データセットを探してみました。

探してみてわかったのは、顔画像データセットって結構少ないということです。

探せば、あることはあるのですが「学術研究用Only」だったり、制約が多かったりして使いにくい。

というわけで、ネットで顔画像データセットを探しました。

Kaggleの顔認証データセット

いろいろ探した結果、Kaggleに使えそうな顔認証のデータセットがあったので紹介します。

ライセンスもCC0(Creative Commons License)でいい感じです。

データセットへのリンクは以下になります。

私はアカウントを持っているので普通にアクセスできましたが、もしかしたら、kaggleにログインしないとアクセスできないかもしれません。

データセットの概要は以下になります。

31人で2562枚なので、人あたり８０枚前後の画像が含まれている計算になります。

データの中身を見ると俳優のBrad Pittの画像など含まれています。肖像権的には不味そうな気もしますが、機械学習に使う分には問題ないはず

枚数的に少し少ない気がしますが、個人の学習や実験用には十分だと思います。とりあえず、顔画像データが見つかったので、「顔認証」の実験にトライしたいと思います。

scikit-learnにも顔画像のデータセットがあります（LFW）。

This dataset is a collection of JPEG pictures of famous people collected over the internet, all details are available on the official website:http://vis-www.cs.umass.edu/lfw/

顔画像は、あらかじめ顔部分が切り出されています。
color=Trueを設定することで３チャネル（カラー）で画像を取り出せます。

顔画像と、IDがペアになったデータセットです。

２枚の顔画像がペアになったデータセットです。

以下のコードで簡単に取り出せるのでちょっとした実験は楽です。

from sklearn.datasets import fetch_lfw_people
lfw_people = fetch_lfw_people(min_faces_per_person=70, resize=0.4)

from sklearn.datasets import fetch_lfw_pairs
lfw_pairs_train = fetch_lfw_pairs(subset='train')

もし、顔画像のデータセットで困っている方がいた場合に参考になればと思い書いてみました。

ここで紹介したデータセットを使って実験を行っています。実験内容については以下のブログ記事を参考にしてください