使用face_recognition人脸检测接口进行人脸检测

import face_recognition
from PIL import Image, ImageDraw
from datetime import datetime

def get_face_detect(path):
    img = face_recognition.load_image_file(path)
    face_locations = face_recognition.api.face_locations(img, 1, 'hog')

    pil_image = Image.fromarray(img)
    draw = ImageDraw.Draw(pil_image)

    # 遍历每个人脸,并标注
    faceNum = len(face_locations)
    for i in range(0, faceNum):
        top = face_locations[i][0]
        right = face_locations[i][1]
        bottom = face_locations[i][2]
        left = face_locations[i][3]

        start = (left, top)
        end = (right, bottom)

        color = (0, 0, 255)
        draw.rectangle((start, end), outline=color)
    del draw
    # 显示结果
    pil_image.show()
    pass
if __name__ == '__main__':
    start_time = datetime.now()
    get_face_detect("0.jpg")
    end_time = datetime.now()
    time_cost = end_time - start_time
    print("TIME_COST:", time_cost)
    pass

2  使用cnn module

face_locations = face_recognition.api.face_locations(img, 1, 'cnn')

猜你喜欢

转载自blog.csdn.net/wss794/article/details/86590900
今日推荐