机器学习之dlib

1)下载dlib
百度网盘地址:https://pan.baidu.com/s/1Z1a_ud__BWXgCWZeSdpL2g
提取码:lzjy
2)下载到本地,我的本地存储位置:C:\Users\lyc\Downloads\dlib-19.7.0-cp36-cp36m-win_amd64.whl

#pycharm下conda激活当前应用的环境
conda activate scikit-learn
#在选择的当前环境下安装dlib
pip install C:\Users\lyc\Downloads\dlib-19.7.0-cp36-cp36m-win_amd64.whl
#安装numpy,matplotlib,opencv-python
pip install numpy
pip install matplotlib
pip install opencv-python

代码应用如下:

import dlib
import cv2
import time

print(dlib.__version__)
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor(r'D:\shape_predictor_68_face_landmarks.dat')
path_screenshots = r"D:"
# 创建 cv2 摄像头对象
cap = cv2.VideoCapture(0)

# 设置视频参数,propId 设置的视频参数,value 设置的参数值
cap.set(3, 960)

# 截图 screenshots 的计数器
ss_cnt = 0

while cap.isOpened():
    flag, img_rd = cap.read()

    # 每帧数据延时 1ms,延时为 0 读取的是静态帧
    k = cv2.waitKey(1)

    # 取灰度
    img_gray = cv2.cvtColor(img_rd, cv2.COLOR_RGB2GRAY)

    # 人脸数
    faces = detector(img_gray, 0)

    # 待会要写的字体
    font = cv2.FONT_HERSHEY_SIMPLEX

    # 按下 'q' 键退出
    if k == ord('q'):
        break
    else:
        # 检测到人脸
        if len(faces) != 0:
            # 记录每次开始写入人脸像素的宽度位置
            faces_start_width = 0

            for face in faces:
                # 绘制矩形框
                cv2.rectangle(img_rd, tuple([face.left(), face.top()]), tuple([face.right(), face.bottom()]),
                              (0, 255, 255), 2)

                height = face.bottom() - face.top()
                width = face.right() - face.left()

                ### 进行人脸裁减 ###
                # 如果没有超出摄像头边界
                if (face.bottom() < 480) and (face.right() < 640) and \
                        ((face.top() + height) < 480) and ((face.left() + width) < 640):
                    # 填充
                    for i in range(height):
                        for j in range(width):
                            img_rd[i][faces_start_width + j] = \
                                img_rd[face.top() + i][face.left() + j]

                # 更新 faces_start_width 的坐标
                faces_start_width += width

            cv2.putText(img_rd, "Faces in all: " + str(len(faces)), (20, 350), font, 0.8, (0, 0, 0), 1, cv2.LINE_AA)

        else:
            # 没有检测到人脸
            cv2.putText(img_rd, "no face", (20, 350), font, 0.8, (0, 0, 0), 1, cv2.LINE_AA)

        # 添加说明
        img_rd = cv2.putText(img_rd, "Press 'S': Screen shot", (20, 400), font, 0.8, (255, 255, 255), 1, cv2.LINE_AA)
        img_rd = cv2.putText(img_rd, "Press 'Q': Quit", (20, 450), font, 0.8, (255, 255, 255), 1, cv2.LINE_AA)

    # 按下 's' 键保存
    if k == ord('s'):
        ss_cnt += 1
        print(path_screenshots + "screenshot" + "_" + str(ss_cnt) + "_" + time.strftime("%Y-%m-%d-%H-%M-%S",
                                                                                        time.localtime()) + ".jpg")
        cv2.imwrite(path_screenshots + "screenshot" + "_" + str(ss_cnt) + "_" + time.strftime("%Y-%m-%d-%H-%M-%S",
                                                                                              time.localtime()) + ".jpg",
                    img_rd)

    cv2.namedWindow("camera", 1)
    cv2.imshow("camera", img_rd)

# 释放摄像头
cap.release()

# 删除建立的窗口
cv2.destroyAllWindows()

在这里插入图片描述

发布了24 篇原创文章 · 获赞 6 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/li_yan_sring/article/details/103538866
今日推荐