ailia_pose_estimator.h

Go to the documentation of this file.

#if !defined(INCLUDED_AILIA_POSE_ESTIMATOR)
#define INCLUDED_AILIA_POSE_ESTIMATOR

/* コアライブラリ */

#include "ailia.h"
#include "ailia_format.h"

/* 呼び出し規約 */

#ifdef __cplusplus
extern "C" {
#endif

/****************************************************************
 * 検出オブジェクトのインスタンス
 **/

struct AILIAPoseEstimator;

/****************************************************************
 * 物体情報
 **/

#define AILIA_POSE_ESTIMATOR_ALGORITHM_ACCULUS_POSE (0)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_ACCULUS_FACE (1)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_ACCULUS_UPPOSE (2)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_ACCULUS_UPPOSE_FPGA (3)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_ACCULUS_HAND (5)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_OPEN_POSE (10)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_LW_HUMAN_POSE (11)

#define AILIA_POSE_ESTIMATOR_ALGORITHM_OPEN_POSE_SINGLE_SCALE (12)

/* 骨格検出 関節点 定義 */
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_NOSE (0)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_EYE_LEFT (1)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_EYE_RIGHT (2)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_EAR_LEFT (3)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_EAR_RIGHT (4)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_SHOULDER_LEFT (5)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_SHOULDER_RIGHT (6)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_ELBOW_LEFT (7)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_ELBOW_RIGHT (8)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_WRIST_LEFT (9)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_WRIST_RIGHT (10)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_HIP_LEFT (11)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_HIP_RIGHT (12)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_KNEE_LEFT (13)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_KNEE_RIGHT (14)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_ANKLE_LEFT (15)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_ANKLE_RIGHT (16)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_SHOULDER_CENTER (17)
#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_BODY_CENTER (18)

#define AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_CNT (19)

/* 顔特徴点検出 定義 */
#define AILIA_POSE_ESTIMATOR_FACE_KEYPOINT_CNT (68)

/* 近接上半身姿勢検出 関節点 定義 */
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_NOSE (0)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_EYE_LEFT (1)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_EYE_RIGHT (2)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_EAR_LEFT (3)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_EAR_RIGHT (4)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_SHOULDER_LEFT (5)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_SHOULDER_RIGHT (6)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_ELBOW_LEFT (7)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_ELBOW_RIGHT (8)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_WRIST_LEFT (9)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_WRIST_RIGHT (10)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_HIP_LEFT (11)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_HIP_RIGHT (12)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_SHOULDER_CENTER (13)
#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_BODY_CENTER (14)

#define AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_CNT (15)

/* 手姿勢検出 点 定義 */
#define AILIA_POSE_ESTIMATOR_HAND_KEYPOINT_CNT (21)

typedef struct _AILIAPoseEstimatorKeypoint {
    float x;
    float y;
    float z_local;
    float score;
    int interpolated;
} AILIAPoseEstimatorKeypoint;

#define AILIA_POSE_ESTIMATOR_OBJECT_POSE_VERSION (1)
typedef struct _AILIAPoseEstimatorObjectPose {
    AILIAPoseEstimatorKeypoint points[AILIA_POSE_ESTIMATOR_POSE_KEYPOINT_CNT];
    float total_score;
    int num_valid_points;
    int id;
    float angle[3];
} AILIAPoseEstimatorObjectPose;

#define AILIA_POSE_ESTIMATOR_OBJECT_FACE_VERSION (1)
typedef struct _AILIAPoseEstimatorObjectFace {
    AILIAPoseEstimatorKeypoint points[AILIA_POSE_ESTIMATOR_FACE_KEYPOINT_CNT];
    float total_score;
} AILIAPoseEstimatorObjectFace;

#define AILIA_POSE_ESTIMATOR_OBJECT_UPPOSE_VERSION (1)
typedef struct _AILIAPoseEstimatorObjectUpPose {
    AILIAPoseEstimatorKeypoint points[AILIA_POSE_ESTIMATOR_UPPOSE_KEYPOINT_CNT];
    float total_score;
    int num_valid_points;
    int id;
    float angle[3];
} AILIAPoseEstimatorObjectUpPose;

#define AILIA_POSE_ESTIMATOR_OBJECT_HAND_VERSION (1)
typedef struct _AILIAPoseEstimatorObjectHand {
    AILIAPoseEstimatorKeypoint points[AILIA_POSE_ESTIMATOR_HAND_KEYPOINT_CNT];
    float total_score;
} AILIAPoseEstimatorObjectHand;

/****************************************************************
 * 骨格検出・顔特徴点検出API
 **/

int AILIA_API ailiaCreatePoseEstimator(struct AILIAPoseEstimator** pose_estimator, struct AILIANetwork* net, unsigned int algorithm);

void AILIA_API ailiaDestroyPoseEstimator(struct AILIAPoseEstimator* pose_estimator);

int AILIA_API ailiaPoseEstimatorSetThreshold(struct AILIAPoseEstimator* pose_estimator, float threshold);

int AILIA_API ailiaPoseEstimatorCompute(struct AILIAPoseEstimator* pose_estimator, const void* src, unsigned int src_stride, unsigned int src_width, unsigned int src_height, unsigned int src_format);

int AILIA_API ailiaPoseEstimatorGetObjectCount(struct AILIAPoseEstimator* pose_estimator, unsigned int* obj_count);

int AILIA_API ailiaPoseEstimatorGetObjectPose(struct AILIAPoseEstimator* pose_estimator, AILIAPoseEstimatorObjectPose* obj, unsigned int obj_idx, unsigned int version);

int AILIA_API ailiaPoseEstimatorGetObjectFace(struct AILIAPoseEstimator* pose_estimator, AILIAPoseEstimatorObjectFace* obj, unsigned int obj_idx, unsigned int version);

int AILIA_API ailiaPoseEstimatorGetObjectUpPose(struct AILIAPoseEstimator* pose_estimator, AILIAPoseEstimatorObjectUpPose* obj, unsigned int obj_idx, unsigned int version);

int AILIA_API ailiaPoseEstimatorGetObjectHand(struct AILIAPoseEstimator* pose_estimator, AILIAPoseEstimatorObjectHand* obj, unsigned int obj_idx, unsigned int version);

#ifdef __cplusplus
}
#endif
#endif /* !defined(INCLUDED_AILIA_POSE_ESTIMATOR) */