python dlib 人脸对齐代码

基于python和dlib编写的人脸对齐程序。包含了两个测试模型（人脸特征点68点检测和5点检测的模型），还有测试图片。详细可以参考博客：http://blog.csdn.net/hongbin_xu/article/details/78511292

# coding: utf-8

# In[2]:


import cv2
import dlib
import sys
import numpy as np
import os


# In[5]:


current_path = os.getcwd（）
# current_path


# In[10]:


predicter_path = current_path + ‘/model/shape_predictor_5_face_landmarks.dat‘# 检测人脸特征点的模型放在当前文件夹中
# predicter_path = current_path + ‘/model/shape_predictor_68_face_landmarks.dat‘
face_file_path = current_path + ‘/faces/inesta.jpg‘# 要使用的图片，图片放在当前文件夹中
print predicter_path
print face_file_path


# In[11]:


detector = dlib.get_frontal_face_detector（）
sp = dlib.shape_predictor（predicter_path）


# In[12]:


bgr_img = cv2.imread（face_file_path）
if bgr_img is None:
    print（“Sorry we could not load ‘{}‘ as an image“.format（face_file_path））
    exit（）


# In[15]:


rgb_img = cv2.cvtColor（bgr_img cv2.COLOR_BGR2RGB）
dets = detector（rgb_img 1）


# In[16]:


num_faces = len（dets）
if num_faces == 0:
    print（“Sorry there were no faces found in ‘{}‘“.format（face_file_path））
    exit（）


# In[18]:


faces = dlib.full_object_detections（）
for det in dets:
    faces.append（sp（rgb_img det））


# In[19]:


images = dlib.get_face_chips（rgb_img faces size=320）
image_cnt = 0
for image in images:
    image_cnt += 1
    cv_rgb_image = np.array（image）.astype（np.uint8）
    cv_bgr_image = cv2.cvtColor（cv_rgb_image cv2.COLOR_RGB2BGR）
    cv2.imshow（‘%s‘%（image_cnt） cv_bgr_image）


# In[20]:


cv2.waitKey（0）
cv2.destroyAllWindows（）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-11-12 13:01  face_alignment\
     目录           0  2017-11-12 12:59  face_alignment\.ipynb_checkpoints\
     文件        3835  2017-11-12 12:58  face_alignment\.ipynb_checkpoints\face_alignment.py-checkpoint.ipynb
     文件        1464  2017-11-12 13:29  face_alignment\face_alignment.py
     文件        3835  2017-11-12 12:59  face_alignment\face_alignment.py.ipynb
     目录           0  2017-11-12 13:11  face_alignment\faces\
     文件       25862  2017-11-12 13:11  face_alignment\faces\inesta.jpg
     文件       11933  2017-09-15 22:42  face_alignment\faces\jobs.jpg
     文件       17660  2017-11-12 13:03  face_alignment\faces\messi.jpg
     文件       35925  2017-09-15 21:19  face_alignment\faces\obama.jpg
     文件       54033  2017-09-15 19:57  face_alignment\faces\silicon_valley.jpg
     目录           0  2017-11-12 13:29  face_alignment\model\
     文件     9150489  2017-11-12 13:29  face_alignment\model\shape_predictor_5_face_landmarks.dat
     文件    99693937  2017-03-09 08:34  face_alignment\model\shape_predictor_68_face_landmarks.dat