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) */