#ifndef __RT_PIXTALKS_FACE_H__ #define __RT_PIXTALKS_FACE_H__ #include "p_code.h" #include "pix_algokit_public_type.h" #include "pix_facekit_api.h" #include "pix_palmkit_api.h" #include "vip_lite.h" #include "aw_util.h" #define VALID_REGISTER_FACE_NUM 1 typedef struct rt_pixtalks_face_config_param { /* model */ unsigned char *detect_model_ptr; unsigned char *face_model_ptr; unsigned char *face_lazy_model_ptr; int detect_model_bytes; int face_model_bytes; int face_lazy_model_bytes; int palm_model_bytes; /* register face info */ pix_image_face_info_t *register_img_face_info[VALID_REGISTER_FACE_NUM]; unsigned char *nir_img_ptr[VALID_REGISTER_FACE_NUM]; unsigned char *rgb_img_ptr[VALID_REGISTER_FACE_NUM]; float pair_fr_score[VALID_REGISTER_FACE_NUM]; }rt_pixtalks_face_config_param; int rt_pixtalks_face_init(int npu_memory_bytes); int rt_pixtalks_face_create(struct rt_pixtalks_face_config_param* face_model_config); int rt_pixtalks_face_cal_fea_sim(struct rt_pixtalks_face_config_param* face_model_config, unsigned char *nir_ptr, unsigned char *rgb_ptr); int rt_pixtalks_face_destroy(struct rt_pixtalks_face_config_param* face_model_config); #endif