《计算机视觉》—— 基于PyCharm中的dlib库实现人脸关键点定位

文章目录

  • [1. 安装必要的库](#1. 安装必要的库)
  • [2. 下载dlib的人脸检测器和关键点预测器模型](#2. 下载dlib的人脸检测器和关键点预测器模型)
  • [3. 编写代码](#3. 编写代码)

人脸关键点定位是指通过计算机视觉技术,识别和定位人脸图像中的关键点,如眼睛、鼻子、嘴巴等特定位置。这些关键点的准确定位对于人脸识别、表情分析、姿态估计等应用具有重要意义。

1. 安装必要的库

  • 首先,你需要确保你的PyCharm环境中安装了dlib、opencv-python(或cv2)和numpy等库。你可以通过PyCharm的Settings -> Project: [Your Project Name] -> Python Interpreter来添加这些库,或者使用pip命令在终端中安装它们:

    bash 复制代码
    pip install dlib opencv-python numpy
  • 在终端里直接安装 dlib 库可能会出现问题,可以参照以下来链接中的内容进行安装

2. 下载dlib的人脸检测器和关键点预测器模型

dlib提供了预训练的人脸检测器和关键点预测器模型。你需要从dlib的官方资源中下载这些模型,并将它们保存到你的项目中。

  • 人脸检测器模型(例如:shape_predictor_68_face_landmarks.dat
  • 人脸识别模型(可选,如果你还需要进行人脸识别)

你可以从dlib的模型下载页面获取这些模型。

3. 编写代码

  • 以下是一个简单的Python脚本,它使用dlib库来加载模型,检测图像中的人脸,并定位关键点:

    python 复制代码
    import numpy as np
    import cv2
    import dlib
    
    # 读取图片
    image = cv2.imread('kobe_2.jpg')    
    
    # 构造人脸检测器
    detector = dlib.get_frontal_face_detector()  
    
    # 检测人脸
    faces = detector(image, 0) 
    
    # dlib.shape_predictor 载入模型(加载预测器)
    # 可以从 https://github.com/davisking/dlib-models 下载 xmlopencv 自己训练好的特征,dlib人脸关键点的检测模型
    predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")
    # 获取每一张脸的关键点(实现检测)
    for face in faces:  
        shape = predictor(image, face)  # 获取关键点
        # 将关键点转换为坐标(x,y)的形式
        landmarks = np.array([[p.x, p.y] for p in shape.parts()])
        # 绘制每一张脸的关键点(绘制shape中的每个点)
        for idx, point in enumerate(landmarks):
            pos = [point[0], point[1]]  # 当前关键点的坐标
            # 针对当前关键点,绘制一个实心圆
            cv2.circle(image, pos, 3, color=(0, 255, 0), thickness=-1)
            # 将每个关键点的索引号写在旁边,             普通大小的等宽字体                                                  线条类型:抗锯齿线条。
            cv2.putText(image, str(idx), pos, cv2.FONT_HERSHEY_SIMPLEX, 0.4, (255, 255, 255), 1, cv2.LINE_AA)
    
    # 显示结果
    cv2.imshow("result", image)
    cv2.waitKey(0)
    # 关闭所有OpenCV创建的窗口
    cv2.destroyAllWindows()
  • 结果如下:

相关推荐
itwangyang5202 分钟前
AIDD-人工智能药物设计-扩散模型热力学:从 AI 提取物理能量
人工智能
ʜᴇɴʀʏ6 分钟前
论文阅读 SAM 3: Segment Anything with Concepts
论文阅读·人工智能·目标检测·计算机视觉·目标跟踪
周杰伦_Jay17 分钟前
【BGE-M3与主流RAG嵌入模型】知识库嵌入模型对比
人工智能·机器学习·eureka·开源·github
Gavin在路上19 分钟前
AI学习之Anthropic的访谈者工具
人工智能·学习
裤裤兔25 分钟前
早停法(Early_Stopping)
人工智能·深度学习
FserSuN33 分钟前
Anthropic文章-打造高性能智能体 学习笔记
人工智能
SaaS_Product33 分钟前
有没有像OneDrive一样的自动同步网盘?
人工智能·云计算·saas·onedrive
我是宝库37 分钟前
Turnitin系统查英文AI率多少为正常?报告显示星号*%怎么办?
人工智能·经验分享·aigc·毕业论文·英文专业·turnitin系统·英文查重
c骑着乌龟追兔子37 分钟前
Day 39 MLP神经网络的训练
人工智能·深度学习·神经网络
infiniteWei40 分钟前
【技术人如何用爬虫+机器学习识别并屏蔽恶意广告】第1课:爬虫与广告反欺诈入门
人工智能·爬虫·机器学习