【计算机视觉】使用python dlib做人脸68点对齐

最新推荐文章于 2023-05-16 18:47:15 发布

usingvc6

最新推荐文章于 2023-05-16 18:47:15 发布

阅读量907

点赞数

分类专栏：计算机视觉文章标签： python 计算机视觉开发语言

本文链接：https://blog.csdn.net/usingvc6/article/details/126037629

版权

计算机视觉专栏收录该内容

2 篇文章 0 订阅

订阅专栏

该博客介绍如何使用Python的dlib库读取文件夹内的图片，并通过预训练的shape_predictor_68_face_landmarks.dat模型进行人脸68点对齐，将结果保存为PTS文件。每个PTS文件包含68个关键点坐标，用于后续的人脸分析或处理。

摘要由CSDN通过智能技术生成

使用python dlib做人脸68点对齐，读取文件夹内image，输出到同目录下同名pts文件
不多说

import os
import dlib
from skimage import io

def landmark_helper(data_dir):

    predictor_path = "model/shape_predictor_68_face_landmarks.dat"

    face_paths= os.listdir(data_dir)

    for path in face_paths:
        face_path = data_dir + path
        '''加载人脸检测器、加载官方提供的模型构建特征提取器'''
        detector = dlib.get_frontal_face_detector()
        predictor = dlib.shape_predictor(predictor_path)

        img = io.imread(face_path)

        dets = detector(img, 1)

        if (len(dets) != 1):
            print('Illegal facial image : ' + face_path + ', detected ' + len(dets) + 'face(s), need 1 !')

        pts_path = face_path.split('.')[0] + '.pts'

        with open(pts_path, mode='w', encoding='utf-8') as file_obj:
            file_obj.write('version: 1\n')
            file_obj.write('n_points:  68\n')
            file_obj.write('{\n')

            for k, d in enumerate(dets):
                shape = predictor(img, d) 
                for p in shape.parts():
                    file_obj.write(str(p.x) + ' ' + str(p.y) + '\n')

            file_obj.write('}\n')

shape_predictor_68_face_landmarks.dat很多地方可以下载，这里贴一个可以下载的gitHub地址