// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
|
// Use of this source code is governed by a BSD-style license that can be
|
// found in the LICENSE file.
|
|
#ifndef EASYMEDIA_RKNN_USER_H_
|
#define EASYMEDIA_RKNN_USER_H_
|
|
#ifdef USE_ROCKFACE
|
#include "rockface/rockface.h"
|
#endif
|
|
#ifdef USE_ROCKX
|
#include <rockx/rockx.h>
|
#endif
|
|
#ifdef __cplusplus
|
extern "C" {
|
#endif
|
|
using RknnCallBack = std::add_pointer<void(void* handler, int type, void *ptr, int size)>::type;
|
// using RknnHandler = std::add_pointer<void*>::type;
|
|
typedef struct Rect {
|
int left;
|
int top;
|
int right;
|
int bottom;
|
} Rect;
|
|
typedef struct {
|
#ifdef USE_ROCKFACE
|
rockface_det_t base;
|
rockface_attribute_t attr;
|
rockface_landmark_t landmark;
|
rockface_angle_t angle;
|
rockface_feature_t feature;
|
#endif
|
#ifdef USE_ROCKX
|
rockx_object_t object;
|
#endif
|
} FaceInfo;
|
|
typedef struct {
|
#ifdef USE_ROCKX
|
rockx_face_landmark_t object;
|
#endif
|
} LandmarkInfo;
|
|
typedef struct {
|
#ifdef USE_ROCKFACE
|
rockface_det_t base;
|
#endif
|
#ifdef USE_ROCKX
|
rockx_keypoints_t object;
|
#endif
|
} BodyInfo;
|
|
typedef struct {
|
#ifdef USE_ROCKX
|
rockx_keypoints_t object;
|
#endif
|
} FingerInfo;
|
|
typedef enum {
|
SUCCESS = 0,
|
FAILURE,
|
TIMEOUT,
|
UNKNOW,
|
} AuthorizedStatus;
|
|
typedef enum {
|
NNRESULT_TYPE_NONE = -1,
|
NNRESULT_TYPE_FACE = 0,
|
NNRESULT_TYPE_FACE_PICTURE_UPLOAD,
|
NNRESULT_TYPE_BODY,
|
NNRESULT_TYPE_FINGER,
|
NNRESULT_TYPE_LANDMARK,
|
NNRESULT_TYPE_AUTHORIZED_STATUS,
|
} RknnResultType;
|
|
typedef struct _RknnResult{
|
int img_w;
|
int img_h;
|
int64_t timeval;
|
RknnResultType type;
|
AuthorizedStatus status;
|
union {
|
BodyInfo body_info;
|
FaceInfo face_info;
|
LandmarkInfo landmark_info;
|
FingerInfo finger_info;
|
};
|
} RknnResult;
|
|
typedef struct _ShmNNData {
|
int32_t size;
|
int64_t timestamp;
|
const char* nn_model_name;
|
RknnResult* rknn_result;
|
} ShmNNData;
|
|
#ifdef __cplusplus
|
}
|
#endif
|
|
#endif // #ifndef EASYMEDIA_RKNN_USER_H_
|
