/*
|
* Copyright 2020 Rockchip Electronics Co. LTD
|
*
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*
|
* author: hh@rock-chips.com
|
* date: 2020-5-19
|
* module: video filter with rknn/rockx/rockface
|
*/
|
#ifndef SRC_RT_MEDIA_AV_FILTER_INCLUDE_RTMEDIAROCKX_H_
|
#define SRC_RT_MEDIA_AV_FILTER_INCLUDE_RTMEDIAROCKX_H_
|
|
#ifdef HAVE_ROCKX
|
#include <rockx/rockx.h>
|
#endif
|
|
#define ROCKX_MODEL_FACE_DETECT "rockx_face_detect"
|
#define ROCKX_MODEL_FACE_LANDMARK "rockx_face_landmark"
|
#define ROCKX_MODEL_POSE_BODY "rockx_pose_body"
|
#define ROCKX_MODEL_POSE_BODY_V2 "rockx_pose_body_v2"
|
#define ROCKX_MODEL_POSE_FINGER "rockx_pose_finger"
|
#define ROCKX_MODEL_FACE_GENDER_AGE "rockx_face_gender_age"
|
|
typedef void (*RTNNCallBack) (void* handler, int32_t type, void *ptr, int32_t size);
|
typedef void* RTNNHandler;
|
|
typedef struct _RTNNInfoFace {
|
#ifdef HAVE_ROCKFACE
|
rockface_det_t base;
|
rockface_attribute_t attr;
|
rockface_landmark_t landmark;
|
rockface_angle_t angle;
|
rockface_feature_t feature;
|
#endif
|
|
#ifdef HAVE_ROCKX
|
rockx_object_t object;
|
#endif
|
} RTNNInfoFace;
|
|
typedef struct _RTNNInfoLandmark {
|
#ifdef HAVE_ROCKX
|
rockx_face_landmark_t object;
|
#endif
|
} RTNNInfoLandmark;
|
|
typedef struct {
|
#ifdef HAVE_ROCKFACE
|
rockface_det_t base;
|
#endif
|
|
#ifdef HAVE_ROCKX
|
rockx_keypoints_t object;
|
#endif
|
} RTNNInfoBody;
|
|
typedef struct _RTNNInfoFinger {
|
#ifdef USE_ROCKX
|
rockx_keypoints_t object;
|
#endif
|
} RTNNInfoFinger;
|
|
typedef enum _RTNNStatus {
|
RT_NN_SUCCESS = 0,
|
RT_NN_FAILURE,
|
RT_NN_TIMEOUT,
|
RT_NN_UNKNOW,
|
} RTNNStatus;
|
|
typedef enum _RTNNDataType {
|
RT_NN_TYPE_NONE = -1,
|
RT_NN_TYPE_FACE = 0,
|
RT_NN_TYPE_BODY,
|
RT_NN_TYPE_FINGER,
|
RT_NN_TYPE_LANDMARK,
|
RT_NN_TYPE_AUTHORIZED_STATUS,
|
} RTNNDataType;
|
|
typedef struct _RTRknnResult {
|
int32_t index;
|
int32_t img_w;
|
int32_t img_h;
|
int64_t timeval;
|
RTNNDataType type;
|
RTNNStatus status;
|
union {
|
RTNNInfoBody info_body;
|
RTNNInfoFace info_face;
|
RTNNInfoLandmark info_landmark;
|
RTNNInfoFinger info_finger;
|
};
|
} RTRknnResult;
|
|
typedef struct _RTRknnAnalysisResults {
|
int32_t count;
|
RTRknnResult *results;
|
} RTRknnAnalysisResults;
|
|
typedef struct {
|
int32_t dataSize;
|
int32_t width;
|
int32_t height;
|
int32_t format;
|
int32_t angle;
|
int32_t mirror;
|
int32_t faceID;
|
unsigned char *feature;
|
int32_t featureLen;
|
} RTKKMattingFaceInfo;
|
|
typedef struct {
|
int32_t faceCount;
|
RTKKMattingFaceInfo *faceInfo;
|
} RTKKAIMattingResult;
|
|
#endif // SRC_RT_MEDIA_AV_FILTER_INCLUDE_RTMEDIAROCKX_H_
|
