armstrong1972 / pysample-for-shiqiyu Goto Github PK

# coding: utf-8
# 
# 程序功能 ：使用 于仕琪 老师的lib，做人脸识别
#     重点 ：调用 C++ 提供的 DLL 接口时，如何处理 指针 类型

import numpy as np
import cv2
from ctypes import *
import types
import sys
import os
import tqdm

Max_Faces = 256
Size_one_Face = 6 + 68 * 2
Size_FaceLandMarks = Size_one_Face * Max_Faces
class FaceResults(Structure): 
    _fields_ = [("face_num", c_int32), 
                ("datas", c_int16 * Size_FaceLandMarks)
               ]

for root, dirs, files in os.walk("E:\\43"):
    for name in files:
        img_path = os.path.join(root, name)
        print(img_path)
        img = cv2.imread(img_path)
        h,w = img.shape[:2]
        st = w*3

        p_img = img.ctypes.data_as(POINTER(c_ubyte))

        #dll = CDLL('./dlls/fd-shiqiyu.dll')    # old version
        dll = CDLL('./dlls/fd-shiqiyu_v2.dll')
        dll.shiqi_fd.restype = POINTER(FaceResults)

        p_results = dll.shiqi_fd(p_img,w,h,st) 
        face_num = p_results.contents.face_num


        for i in range(face_num):
            j =  Size_one_Face * i
            x = p_results.contents.datas[j]
            y = p_results.contents.datas[j+1]
            w = p_results.contents.datas[j+2]
            h = p_results.contents.datas[j+3]
            confidence = p_results.contents.datas[j+4]
            angle = p_results.contents.datas[j+5]