#!/usr/bin/env python3

#-*- coding:utf-8 -*-# __author__:kzg'''cv2.imshow("image", img_blank) # 显示图片

cv2.waitKey(0) # 窗口一直开着'''import cv2

import dlib

import numpyasnp

import os, sys, timefromrandom import randint

BASEDIR=os.path.dirname(os.path.abspath(__file__))

# 图片源

sourceDir= os.path.join(BASEDIR, "pic_folder")

# 保存使用过的文件

dataset= os.path.join(BASEDIR, 'dataset')

# 处理图片保存目录

outPic= os.path.join(BASEDIR, "extra_folder")

def tranposeDector(img, cnt=0):''':param infile: imread后的图片对象'''# 检测图上的人脸数try:

dets= dector(img, 1)

except Exceptionasex: # 脸部无法检测return(ex)

# 身份证上只能有一个人脸，即为检查结果的第一个值ifdets:

face= dets[0] # [(354, 96) (444, 186)] 检测出左上、右下两个点

# 计算想裁取的图片的高度 下-上

height= face.bottom() - face.top() + 60# 计算想裁取的图片的宽度 右-左

width= face.right() - face.left() + 40# 以计算出的图片大小生成空白板

img_blank= np.zeros((height, width, 3), np.uint8)

# 将图片写入空白板try:for i inrange(height):for j inrange(width): # top线上方40像素位置开始读, left线左15像素位置开始读

img_blank[i][j]= img[face.top() - 40 + i][face.left() - 15 +j]

cv2.imwrite(os.path.join(outPic, str(randint(0,10000))+".jpg"), img_blank) # 保存写入数据后的空白板图片

except Exceptionasex:

print(ex)

cv2.destroyAllWindows() # 释放所有窗口资源else:

cnt+= 1

if cnt < 3:

transposeImage=cv2.transpose(img) # 图像反向旋转90度

flipedImageX= cv2.flip(transposeImage, 0) # 沿X轴方向的镜像图片

cv2.imshow("flipedImageX",flipedImageX)

cv2.waitKey(1000)

tranposeDector(flipedImageX, cnt)else:

print("人脸检测失败 transpose times:", cnt)

cv2.destroyAllWindows() # 释放所有窗口资源if __name__ == '__main__':

# 初始化正脸检测器

dector=dlib.get_frontal_face_detector()whileTrue:

# 图片目录下所有文件集(增加文件后缀判断)

sSet= set(os.listdir(sourceDir))

# 数据集中所有文件名ifos.path.exists(dataset):

with open(dataset,'r') asfp:

dSet= set()for i infp.readlines():

dSet.add(i.replace("\n", ""))else:

with open(dataset,'w') asfp:

fp.write("")

fp.close()

dSet= set()

sSet.difference_update(dSet)for p insSet:

print("start:", p)

# 利用opencv读取图片

img=cv2.imread(os.path.join(sourceDir, p))

tranposeDector(img)

# 转换过的图片保存到dataset

with open(dataset,'a') asfp:

fp.write(p+ "\n")

time.sleep(5)