PyTorch人脸检测

news2024/9/27 21:18:18

新书速览|PyTorch深度学习与企业级项目实战-CSDN博客

人脸检测解决的问题是确定一幅图上有没有人脸，而人脸识别解决的问题是这张脸是谁的。可以说人脸检测是人脸识别的前期工作。这里介绍Dlib库，它提供了Python接口，里面有人脸检测器，有训练好的人脸关键点检测器。

安装OpenCV，可以通过下载OpenCV的.whl文件，使用pip install opencv_python-3.4.0-cp36-
cp36m-win_amd64.whl命令来安装。如果import cv2报错ImportError: numpy.core.multiarray failed to import，出现这个问题的解决方法是下载最新版本的NumPy，解决方法是输入命令pip install numpy -–upgrade，结果如图11-3所示表示成功解决。

图11-3

进行实时图像捕获，首先需要学点OpenCV的基础知识，起码知道如何从摄像头获取当前拍到的图像。本项目使用Dlib库，Dlib库提供的功能十分丰富，它提供了Python接口，里面有人脸检测器，也有训练好的人脸关键点检测器。

程序代码如下：

##########实时检测视频中的人脸###############################
import cv2
import dlib
predictor_path = ".\shape_predictor_68_face_landmarks.dat"

#使用dlib自带的frontal_face_detector作为人脸检测器
detector = dlib.get_frontal_face_detector()
# 使用官方提供的模型构建特征提取器
predictor = dlib.shape_predictor(predictor_path)

#初始化窗口
win = dlib.image_window()
cap = cv2.VideoCapture(0) 	#获取摄像头
while cap.isOpened():  		#读取摄像头的图像，函数 isOpened用于判断摄像头是否开启
    ok,cv_img = cap.read()  
    img = cv2.cvtColor(cv_img, cv2.COLOR_RGB2BGR)　#转灰度化，简化图像信息
 
   # 与人脸检测程序相同，使用detector进行人脸检测，dets为返回的结果
    dets = detector(img, 0)
shapes =[]
if cv2.waitKey(1) & 0xFF == ord('q'):
    print("q pressed")
    break
else:
    # 使用enumerate 函数遍历序列中的元素以及它们的下标
    # 下标k即为人脸序号
    for k, d in enumerate(dets):
        # 使用predictor进行人脸关键点识别， shape为返回的结果 
        shape = predictor(img, d)
        #绘制特征点
        for index, pt in enumerate(shape.parts()):
            pt_pos = (pt.x, pt.y)
            cv2.circle(img, pt_pos, 1, (0,225, 0),2) #利用cv2.putText输出1-68
            font = cv2.FONT_HERSHEY_SIMPLEX
            cv2.putText(img, str(index+1),pt_pos,font,
                        0.3, (0, 0, 255), 1, cv2.LINE_AA)
    win.clear_overlay()
    win.set_image(img)
    if len(shapes)!= 0 :
        for i in range(len(shapes)):
            win.add_overlay(shapes[i])        
    win.add_overlay(dets)       

cap.release()
cv2.destroyAllWindows()

运行结果如图11-4所示。