data_weibo_process.py

# -*- coding: utf-8 -*-
"""
Created on Sun Apr 12 14:29:55 2020

@author: dell
"""

# -*- codeing: utf-8 -*-
import sys
import os
import cv2
import dlib

#准备好的数据集 '../zhengshuang_weibo'
input_dir = './raw_data/liuchang'

#输出路径
output_dir = './out_data/liuchang'
size = 128

if not os.path.exists(output_dir):
    os.makedirs(output_dir)

#使用dlib自带的frontal_face_detector作为我们的特征提取器
detector = dlib.get_frontal_face_detector()

index = 1
for (path, dirnames, filenames) in os.walk(input_dir):
    for filename in filenames:
        if filename.endswith('.jpg'):
            print('Being processed picture %s' % index)
            img_path = path+'/'+filename
            # 从文件读取图片
            img = cv2.imread(img_path)
            # 转为灰度图片
            gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
            # 使用detector进行人脸检测 dets为返回的结果
            dets = detector(gray_img, 1)

            #使用enumerate 函数遍历序列中的元素以及它们的下标
            #i为索引，d为值
            #left：人脸左边距离图片左边界的距离 ；right：人脸右边距离图片左边界的距离 
            #top：人脸上边距离图片上边界的距离 ；bottom：人脸下边距离图片上边界的距离
            for i, d in enumerate(dets):
                top = d.top() if d.top() > 0 else 0
                bottom = d.bottom() if d.bottom() > 0 else 0
                left = d.left() if d.left() > 0 else 0
                right = d.right() if d.right() > 0 else 0
                # img[y:y+h,x:x+w]
                face = img[top:bottom,left:right]
                # 调整图片的尺寸至64*64
                face = cv2.resize(face, (size,size))
                cv2.imshow('image',face)
                # 保存图片
                cv2.imwrite(output_dir+'/'+str(index)+'.jpg', face)
                index += 1

            key = cv2.waitKey(30) & 0xff
            if key == 27:
                sys.exit(0)