38 lines
1.2 KiB
C
38 lines
1.2 KiB
C
|
#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
|