(dlib) Python

Anacondaで導入してFace landmarks抽出のテスト

インストール

Anacondaで入れる

conda install --channel https://conda.anaconda.org/menpo dlib

テスト

Anacondaで入れたのはX11がらみが入ってないのでmatplotlibで代用。 学習済みのpredictorを持ってくる

wget http://sourceforge.net/projects/dclib/files/dlib/v18.10/shape_predictor_68_face_landmarks.dat.bz2
bzip2 -d shape_predictor_68_face_landmarks.dat.bz2

テストはlenaで。

import dlib
from skimage import io
import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
from matplotlib.patches import Circle

detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor('shape_predictor_68_face_landmarks.dat')
img = io.imread('../lena.jpg')
dets = detector(img, 1)

plt.imshow(img)
ax = plt.gca()
ax.add_patch(Rectangle((dets[0].top(),dets[0].left()),dets[0].width(),dets[0].height(),fill=False))
shape = predictor(img, dets[0])
for i in range(shape.num_parts):
    ax.add_patch(Circle((shape.part(i).x,shape.part(i).y), radius=1.0, fc='y'))
plt.show()

お手軽。