python 识别图片中的人脸，并裁剪人脸，将图片转化成固定大小，并将图片再按照自己的后缀名存储起来

下列代码，识别图片中的人脸，并裁剪人脸，将图片转化成固定大小，再按照自己的后缀名存储起来

bug:我是用来识别单张图片中只有一个人脸的数据集，如果用来处理一张图片中的多个人脸，那么图片的命名可能会有bug

import cv2

import os

import glob

size_m = 48

size_n = 48

def detect(img, cascade):

rects = cascade.detectMultiScale(img, scaleFactor=1.3, minNeighbors=4, minSize=(30, 30),flags=cv2.CASCADE_SCALE_IMAGE)

if len(rects) == 0:

return []

rects[:, 2:] += rects[:, :2]

return rects

cascade = cv2.CascadeClassifier("cascades\\haarcascade_frontalface_alt2.xml")

imglist=glob.glob("data/image/jaffe/*")

for list in imglist:

img = cv2.imread(list)

dst=img

rects=detect(dst,cascade)

for x1,y1,x2,y2 in rects:

roi = dst[y1+10 :y2+20, x1+10 :x2 ]

img_roi = roi

re_roi = cv2.resize(img_roi, (size_m, size_n))

f = "{}/{}".format("data/image", "jaffe_1")

if not os.path.exists(f):

os.mkdir(f)

path=list.split(".")

cv2.imwrite("{}/{}_{}.jpg".format(f,path[1],path[2]), re_roi)