一种简单人脸识别定位的实现

#coding=utf-8
import facekit_reina
import os 
#未处理图片存放路径
raw_path = os.getcwd()+'/'
#处理后图片路径
positioned_path = raw_path + 'positioned/'

facekit_reina.start_generation(raw_path, positioned_path)

#coding=utf-8
import face_recognition
from cv2 import cv2
import os

# generate for each image
# 每个文件生成图片
def generate_face_positioned_img(path, filename, positioned_path):
    image = face_recognition.load_image_file(path + filename)
    face_locations=face_recognition.face_locations(image)
   
    # face_locations eg:(139,283,325,97) (y1,x1,y2,x2)
    # reference of coordinate
    # 坐标系参考
    # ######################---------→ x
    # #                    #
    # #                    #
    # #                    #
    # #                    #
    # #       IMAGE        #
    # #                    #
    # #                    #
    # #                    #
    # #                    #
    # ######################
    # |
    # |
    # |
    # ↓
    # y

    for face in face_locations:
      y1=face[0]
      x1=face[1]
      y2=face[2]
      x2=face[3]
      xmin = min(x1,x2)
      ymin = min(y1,y2)
      xmax = max(x1,x2)
      ymax = max(y1,y2)
      print(xmin,xmax,ymin,ymax)
      cv2.rectangle(image,(xmin,ymin),(xmax,ymax),(255,0,0),3)

    #os.mkdir(positioned_path)
    cv2.imwrite(positioned_path+filename,cv2.cvtColor(image, cv2.COLOR_RGB2BGR))

# flow control
# 流程控制
def start_generation(raw_image_path, positioned_image_path):
    
    # directory of source files
    # 源文件的目录
    #raw_image_path='src/known/'

    # directory of output files
    # 输出文件的目录
    #positioned_image_path=raw_image_path+'positioned/'

    # filename List under the path (all file and floder included)
    # 文件名的列表 (包括所有的文件和文件夹)
    image_name_list=os.listdir(raw_image_path)

    if not os.path.exists(positioned_image_path):
        os.mkdir(positioned_image_path)

    # Compatible Formats  
    # 支持的文件格式
    compatible_formats=['.jpg','.jpeg','.gif','.png']

    all_overwrite = False

    for ifilename in image_name_list:
        if os.path.isdir(raw_image_path + ifilename):
            continue
        print("processing " + ifilename + ' ...')
        if (os.path.splitext(ifilename)[1] in compatible_formats):
            if (ifilename not in os.listdir(positioned_image_path)):
                generate_face_positioned_img(raw_image_path, ifilename, positioned_image_path)
            else:
                if all_overwrite == True:
                    pass
                else:
                    while True:
                        is_overwrite = input(ifilename+" already exists, do you want to overwrite it? (Y/n/all): ")
                        if str.lower(is_overwrite) == 'y':
                            generate_face_positioned_img(raw_image_path, ifilename, positioned_image_path)
                            break
                        elif str.lower(is_overwrite) == 'n':
                            break
                        elif str.lower(is_overwrite) == 'all':
                            generate_face_positioned_img(raw_image_path, ifilename, positioned_image_path)
                            all_overwrite = True
                            break
                        else:
                            pass
        else:
            print("Format not supported!")
            while True:
                is_show_compatible_formats = input('Show all supported formats? (Y/n): ')
                if str.lower(is_show_compatible_formats) == 'y':
                    for one_format in compatible_formats:
                        print(one_format)
                    break
                elif str.lower(is_show_compatible_formats) == 'n':
                        break
                else:
                    pass

    print("All completed!")