顔検出を実装するための50行のPythonコード

現在、顔認識技術は広く利用されており、決済や本人認証、美容カメラなどに利用されています。 iPhone ユーザーは、次の機能に精通している必要があります。

iPhoneの写真には、写真に写っている顔を認識して分類できる「People」機能があります。その原理も顔認識技術です。

この記事では主にPythonを使って顔検出を実装する方法を紹介します。顔検出は顔認識の基礎です。顔検出の目的は、写真に写っている顔を識別し、顔の特徴を見つけることです。顔認識は、顔検出に基づいて、その人物が誰であるかをさらに知らせることです。

さて、紹介はこれで終わりです。次に、環境の準備を始めましょう。

準備

この記事の顔検出は dlib に基づいていますが、これは Boost と cmake に依存しているため、まずこれらのパッケージをインストールする必要があります。Ubuntu を例に挙げます。

 $ sudo apt-get install build-essential cmake 
 
 $ sudo apt-get で libgtk-3-dev をインストールします
 
 $ sudo apt-get install libboost- all -dev

私たちのプログラムは numpy と opencv も使用するため、これらのライブラリもインストールする必要があります。

 $ pip numpyをインストール
 
 $ pip scipyをインストール
 
 $ pip opencv-pythonをインストール
 
 $ pip dlibをインストール

顔検出は、ここからダウンロードできる事前トレーニング済みのモデルデータに基づいています。

 http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2

ローカルパスにダウンロードした後、解凍して、プログラムで使用される解凍されたファイルのパスを書き留めます。

dlibの顔の特徴

上記でダウンロードしたモデルデータを使用して、顔の68個の特徴点（x、y）の座標位置を推定します。これらの68個の座標点の位置は次の図に示されています。

手順は次の 2 つのステップで構成されます。

最初のステップは写真の顔の部分を検出することです

2 番目のステップでは、検出された顔領域内の器官 (目、鼻、口、あご、眉毛) をさらに検出します。

顔検出コード

まず、いくつかのユーティリティ関数を定義しましょう。

 rect_to_bb(rect)を定義します。 
 
    x = rect.left () 
 
    y =矩形上端() 
 
    w = rect.right () - x 
 
    h = rect.bottom() - y 
 
 (x, y, w, h)を返す

この関数の rect は、dlib 顔領域検出の出力です。ここで、rect はシーケンスに変換され、その内容は矩形領域の境界情報となります。

 def shape_to_np(shape, dtype= "int" ): 
 
    座標 = np.zeros((68, 2), dtype=dtype) 
 
 iが範囲(0, 68)内にある場合: 
 
            座標[i] = (shape.part(i).x, shape.part(i).y) 
 
座標を返す

この関数のシェイプは、dlib 顔特徴検出の出力です。シェイプには、上記の顔特徴の 68 個のポイントが含まれます。この関数は、後続の処理を容易にするために、形状を Numpy 配列に変換します。

 def resize(画像、幅=1200): 
 
    r = 幅 * 1.0 / image.shape[1] 
 
    dim = (幅、 int (image.shape[0] * r)) 
 
    サイズ変更 = cv2.resize(画像、dim、補間=cv2.INTER_AREA) 
 
サイズ変更して返す

この関数内の画像は、検出したい画像です。顔検出プログラムの最初に、確認のために検出結果の画像を表示します。ここでサイズを変更するのは、画像が大きくなりすぎて画面からはみ出ないようにするためです。

次に、メインプログラムを開始しましょう。

 sysをインポートし、numpyをnpとしてインポートします。 
 
 dlibをインポート cv2をインポート
 
 len(sys.argv) < 2の場合: 
 
    print "使用法: %s <画像ファイル>" % sys.argv[0] 
 
    sys.exit(1) 
 
イメージファイル = sys.argv[1] 
 
検出器 = dlib.get_frontal_face_detector() 
 
予測子 = dlib.shape_predictor( "shape_predictor_68_face_landmarks.dat" )

顔を検出する画像をsys.argv[1]パラメータから読み取り、顔領域検出用の検出器と顔特徴検出用の予測子を初期化します。 shape_predictor のパラメーターは、先ほど解凍したファイルへのパスです。

画像 = cv2.imread(画像ファイル) 
 
画像 = resize(画像、幅=1200) 
 
グレー = cv2.cvtColor(画像、cv2.COLOR_BGR2GRAY) 
 
 rects = detector(グレー, 1)

特徴領域を検出する前に、まず顔領域を検出する必要があります。このコードは opencv を呼び出して画像を読み込み、適切なサイズに変更し、グレースケールに変換し、最後に detector を使用して顔領域を検出します。写真には複数の顔が含まれている可能性があるため、ここでは複数の顔に関する情報を含む配列 rects を取得します。

 enumerate(rects)内の(i, rect)について: 
 
    形状 = 予測子(灰色、四角形) 
 
    形状 = shape_to_np(形状) 
 
    (x, y, w, h) = rect_to_bb(rect) 
 
    cv2.rectangle(画像、(x, y)、(x + w, y + h)、(0, 255, 0)、2) 
 
    cv2.putText(image, "Face #{}" .format(i + 1), (x - 10, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2) 
 
 (x, y)の形状は次のようになります。 
 
            cv2.circle(画像, (x, y), 2, (0, 0, 255), -1) 
 
 cv2.imshow( "出力" , 画像) 
 
 cv2.waitKey(0)