数据预处理之提取人脸关键点并保存至.csv文件中

最新推荐文章于 2023-07-04 10:09:32 发布

松鼠爆米花

最新推荐文章于 2023-07-04 10:09:32 发布

阅读量1.2k

点赞数 2

分类专栏： python 数据预处理

本文链接：https://blog.csdn.net/leonorski/article/details/105562429

版权

这篇博客记录了如何从数据集中提取人脸关键点，并将这些信息保存到CSV文件中，每一行包含一个人脸的关键点信息。

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

经常看到有代码中将数据集中人脸的关键点提取出来，并放在csv文件中，

csv中每行存放一张人脸的信息，格式为：

image_name,part_0_x,part_0_y,part_1_x,part_1_y,...,part_67_x,part_67_y

看到网上一个不错的实现方法，故记录一下。

首先是 create_csv.py

import dlib
import glob
import cv2
import csv
from skimage import io
import os

root_dir = '/remote-home/my/xxxxx'

detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor('shape_predictor_68_face_landmarks.dat')
num_landmarks = 68

with open('train_cords.csv', 'w', newline='') as csvfile:
    csv_writer = csv.writer(csvfile)

    header = ['image_name']
    for i in range(num_landmarks):
        header += ['part_{}_x'.format(i), 'part_{}_y'.format(i)]
    csv_writer.writerow(header)

    f = open('train_file.csv')
    imgs = f.readlines()