LoginSignup
7
7

More than 5 years have passed since last update.

opencv4nodejs, face-recognition.js で顔認識のプログラムを開発してみよう

Last updated at Posted at 2019-02-17

対象読者

  • NodeJSの言語を開発しOpenCVのライブラリを使いたい人です。
  • 顔認識というプログラムを作りたい人です。

目標

  • 基本的な顔認識のプログラムを開発する方法がわかって開発するようになります。
  • 機械学習(Machine Learning)のコンピュータビジョンの基本的な結構とイメージの処理の部分がわかります。

筆者

  • コンピュータビジョン初心者。最近イメージの処理のテクノロジーが素晴らしいです。
  • まだまだ分からないことが多いので、間違いの指摘などよろしくお願いします。

1.はじめに

まず、プログラムの稼働を説明させていただきます。そのプログラムは顔を認識することができるというプログラムです。カメラを見るときにビデオにその人の顔と名前を表示します。カメラを見る人がたくさんいるも大丈夫です。新顔をカメラを見る場合はその人の顔のイメージを集めて次回はその人の人を発見することができます。
以上の性能を記述してからプログラムの設計を見ましょう。
Screenshot from 2019-02-17 15-53-11.png
以上のダイヤグラムを見るとプログラムの動く方法がわかると思います。プログラムは主な機能が2つあります。顔のデータがある場合は認識することができて新顔の場合はユーザーが名前をインプットします。それからウエブカメラを呼び出してフレームを取得してイメージを処理します。ビデオというのはイメージがいっぱい接続します。だから私たちはビデオを処理のかわりにイメージ(ビデオのフレーム)で処理しています。

2.前提知識

プログラムの動く方法がわかった上で必要な知識をまとめます。

機械学習についてトレーニングのグラフを見て見ましょう。
Screenshot from 2019-02-17 15-33-22.png
まずはトレーニングのステプです。誰かの顔のイメージをたくさん集めてトレニングしてモデルファイルを作ります。各人の顔は、画像処理技術の抽出の特徴的な方法に基づく特徴ベクトルによって描かれます。それでトレーニングファイルが作られます。そのファイルで顔認識するときにそれぞれの人の顔のデータを取得することができます。

言語についてNodeJSとTypeScriptの基本的な使い方がわかることは大事です。それからopencv4nodejsのライブラリをインストールしてドキュメントを読むすることが必要です。そしてライブラリの中に例がありますから使い方をわかりやすいになると思います。次はface-recognitionのライブラリを調べたほうがいいと思います。そのライブラリのドキュメントは本当に役に立ちます。それはライブラリの関数でフレームから顔の認識する方法があり精度レベールもあります。例えば以下の説明を見えます。
Screenshot from 2019-02-17 15-15-05.png
それぞれのフレームに対して[predict]の関数を使ってモデルファイルのそれぞれの顔を比較して似ているレベールが取得することができます。誰の顔が最もビデオに似ているか知りたい場合は[predictBest]の関数を使います。説明もあるし例もあるので難しくないと思います。

3.コードを読んでいく

3.1. 顔を認識する

この機能をもっと小さな関数に分割します。 このように分割すると、コードがよりきれいになり、保守しやすくなります。
最初は顔を検出する関数です。イメージからopencvのhaar_cascadeという分類のアルゴリズムで顔検出します。出す結果は顔の周りに描かれた長方形が識別されます。以下のイメージはコードです。
Screenshot from 2019-02-17 15-47-14.png
それぞれ検出した顔に応じてフレームに長方形を描きます。その関数は簡単です。ライブラリの2つの関数を使って長方形を描く関数とテキストを書く関数です。
Screenshot from 2019-02-17 16-18-53.png
次はウエブカメラを呼び出します。その関数はopencv4nodejsのライブラリのドキュメントがありますから簡単になります。ちょっとコードに説明します。ここでウエブカメラの各フレームを読み取るためのループを作成し、各フレームでonFrame関数を処理します。 キーが押されると、ループはクリアされて停止し、プログラムは終了します。
Screenshot from 2019-02-17 16-06-25.png
それからフレームの処理のステプです。ちょっと長いのでもっと小さに分割させていただきます。
Screenshot from 2019-02-17 16-14-40.png
ほとんどはわかりやすいです。ちょっと説明してあげます。ここにthresholdの変数は最低の似ているレベールです。顔認識するときにface-recognitionのライブラリで2つの結果を取得します。クラスネーム(人の名前)を応じて間違っているレベールがあります。間違っているレベールが大きいほど、実際の顔とデータベース内の顔との間の差が大きくなります。それが最大許容値を超えると、それは受け入れられず、新しい顔として識別されます。ほかにはvarsTimeIn、varsTimeOutとflagsDetectの変数はタイムスタンプ変数と顔検出のフラグ変数です。顔検出する場合は長方形を描きます。その顔がカメラを見る時間を保存されています。
Screenshot from 2019-02-17 16-37-27.png
反対の場合10秒あとでその人がカメラを見ないとスタート時間とエンド時間の結果がログされています。時々、外部の要素によってプログラムが私たちの顔をしばらく認識できなくなるでしょう。そしてどんな理由でカメラを短い時間で見なくてもいいです。以上のコードでその問題を解決しました。

3.2. 新顔を保存する

知らない人がカメラを見る場合は以上に説明したthresholdの変数でその人の顔の下に[unknown]というテキストがあります。だからまずはトレーニングしなければなりません。
トレーニングのために3つのファイルを準備します。これはNodeJSでファイルを処理するようなファイルです。
最初はパスをチェックするファイル。NodeJSの[path]と[filesystem(fs)]を使ってシステムにデータフォルダが存在することを確認します。ここに「データ」と「モデル」というフォルダをチェックしてなっかた場合は新しいフォルダを作成します。
Screenshot from 2019-02-23 18-47-12.png
次は新規なイメージを作成するファイルであります。人の名前や索引やフレームから新規な顔のイメージ作られます。
Screenshot from 2019-02-23 18-56-32.png
あとはトレーニングファイルです。以上の方法で作ったイメージを利用して新顔をトレーニングします。簡単に言えば、それはこの新しい顔のデータをファイルモデルに追加することです。以下のイメージでちょっと長いですがかんたんですぐに説明します。最初は顔のイメージのフォルダに新人のイメージを全部検索してface-recognitionのライブラリの「addFaces」という関数を利用して新顔のデータを含むJSONを作られてモデルファイルに追加します。
Screenshot from 2019-02-23 19-03-05.png
以下の3つのファイルがあったら次は主なトレーニングの機能を開発しています。
フレームを取得する方法は顔認識の機能にほとんど同じですがちょっと変更します。はじめに「grabFramesTraining」という関数を作ります。
Screenshot from 2019-02-23 19-16-28.png
ここで顔認識の部分の「grabFrames」の関数とちょっと違います。関数のインプットは新人の名前を追加することが大切です。キーボードのキーを押すと現在のフレームに顔があるかどうか確認してある場合はそのフレームを保存してない場合は保存しません。ここでトレーニングのため保存するイメージの数は10ですが実際、必要な画像の数は非常に多くなります。これは、画像の数が多いほど、データが多くなり、識別の精度が高くなるためです。10枚の写真がある場合は、上記で作成したトレーニングファイルからトレーニングを実行します。
次に「runVideoNewFaceTraining」を作成します。
Screenshot from 2019-02-23 21-10-43.png
ほとんど以上のコードは顔認識の部分と同じです。少し違うところは「check_face」という変数が出現します。その変数はなんのフレームを保存するときにそのフレームに顔が検出するかどうか確認する変数です。他にはユーザーに質問を出るためにNodeJSのreadlineを利用します。

3.3. 主なプログラムを完成する

この部分はNodeJSのreadlineを利用して、以上の2つの機能を呼び出します。コードは以下のイメージのようにします。
Screenshot from 2019-02-23 21-54-08.png
これまで全部顔認識のプログラムを完了しました。

4.デモする

プログラム実行時の画像は以下のとおりです。

プログラムを実行する画面

Screenshot from 2019-02-23 22-12-37.png

新顔をトレーニングをする画面

Screenshot from 2019-02-23 22-12-50.png

完了する画面

Screenshot from 2019-02-23 22-13-28.png

顔認識を実行する画面

Screenshot from 2019-02-23 22-13-55.png

二人(ここで私とスマートフォンからのイメージ)がカメラを見る画面

Screenshot from 2019-02-23 22-22-03.png

続いてhowardさんが出て私もカメラを見てhowardさんのタイミングをログする画面

Screenshot from 2019-02-23 22-22-50.png

私も出てプログラムを停止して二人の時間をタイミングをログする画面

Screenshot from 2019-02-23 22-23-26.png

最後に

最後まで私たちは簡単な顔認識プログラムを作りました。 読んでいただきありがとうございます、あなたが理解できない何かがあれば、私にメールで連絡してください。ソースコードの詳細は以下のGitHubのリポジトリで見てください。
Email: hungph.dev.ict@gmail.com
Github: https://github.com/hungph-dev-ict/face_detect_app

7
7
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
7
7