rknn调用人脸属性api

流程为先人脸检测，然后人脸对齐，然后人脸属性识别

import cv2
import time
from rockx import RockX
import numpy as np
ROCKX_DATA_DIR = "/home/coolpi/RockX_SDK_V1.6.0_20220913/sdk/rockx-data-rk3588"
FACE_DETECT_MIN_SCORE = 0.6
#存放了模型和数据
handle_face = RockX(RockX.ROCKX_MODULE_FACE_DETECTION, rockx_data_dir=ROCKX_DATA_DIR)
handle_face_attribute = RockX(RockX.ROCKX_MODULE_FACE_ATTRIBUTE, rockx_data_dir=ROCKX_DATA_DIR)
face_landmark5_handle = RockX(RockX.ROCKX_MODULE_FACE_LANDMARK_5, rockx_data_dir=ROCKX_DATA_DIR)

#调用RockX类的rockx_person_detect方法
process_face = getattr(handle_face, "rockx_face_detect")
process_face_attribute = getattr(handle_face_attribute, "rockx_face_attribute") #rockx_pose_body
facepath = "dataset/run/"
dataset = 'dataset/stand1.mp4'
cap = cv2.VideoCapture(dataset)
flag, frame = cap.read()
j = 0
det_results = [] 
while flag:
    # 读取每一帧，flag表示是否读取成功，frame为图片内容。
    flag, frame = cap.read()
    # frame = cv2.pyrDown(frame)
    if frame is None:
        # print("No video captured")
        continue
    else:
        width = frame.shape[1]
        height = frame.shape[0]
        ################
        ret, results = process_face(
            frame, width, height,RockX.ROCKX_PIXEL_FORMAT_BGR888)       
        for result in results:
            if result.score >FACE_DETECT_MIN_SCORE:
                b = [result.box.top, result.box.right, result.box.bottom, result.box.left]
                # ret, face = process_face(frame, width, height, RockX.ROCKX_PIXEL_FORMAT_BGR888, result.box)
                ret, align_img = face_landmark5_handle.rockx_face_align(frame, width, height,
                                                                             RockX.ROCKX_PIXEL_FORMAT_BGR888,
                                                                             result.box, None)
                ret, face_attr = process_face_attribute(align_img)
                print(face_attr)
                cv2.imwrite(facepath+str(j)+'.jpg', frame)
                j=j+1
                print(j)