ailia_classifier.h

Go to the documentation of this file.

#if !defined(INCLUDED_AILIA_CLASSIFIER)
#define INCLUDED_AILIA_CLASSIFIER

/* コアライブラリ */

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

/* 呼び出し規約 */

#ifdef __cplusplus
extern "C" {
#endif

/****************************************************************
 * 識別オブジェクトのインスタンス
 **/

struct AILIAClassifier;

/****************************************************************
 * 識別情報
 **/

#define AILIA_CLASSIFIER_CLASS_VERSION (1)

typedef struct _AILIAClassifierClass {
    int category;
    float prob;
} AILIAClassifierClass;

int AILIA_API ailiaCreateClassifier(struct AILIAClassifier** classifier, struct AILIANetwork* net, unsigned int format, unsigned int channel, unsigned int range);

void AILIA_API ailiaDestroyClassifier(struct AILIAClassifier* classifier);

int AILIA_API ailiaClassifierCompute(struct AILIAClassifier* classifier, const void* src, unsigned int src_stride, unsigned int src_width, unsigned int src_height, unsigned int src_format, unsigned int max_class_count);

int AILIA_API ailiaClassifierGetClassCount(struct AILIAClassifier* classifier, unsigned int* cls_count);

int AILIA_API ailiaClassifierGetClass(struct AILIAClassifier* classifier, AILIAClassifierClass* obj, unsigned int cls_idx, unsigned int version);

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