ailia_detector.h

Go to the documentation of this file.

#if !defined(INCLUDED_AILIA_DETECTOR)
#define INCLUDED_AILIA_DETECTOR

/* コアライブラリ */

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

/* 呼び出し規約 */

#ifdef __cplusplus
extern "C" {
#endif

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

struct AILIADetector;

/****************************************************************
 * 物体情報
 **/
#define AILIA_DETECTOR_OBJECT_VERSION (1)

typedef struct _AILIADetectorObject {
    unsigned int category;
    float prob;
    float x;
    float y;
    float w;
    float h;
} AILIADetectorObject;

#define AILIA_DETECTOR_ALGORITHM_YOLOV1 (0)  // YOLOV1
#define AILIA_DETECTOR_ALGORITHM_YOLOV2 (1)  // YOLOV2
#define AILIA_DETECTOR_ALGORITHM_YOLOV3 (2)  // YOLOV3
#define AILIA_DETECTOR_ALGORITHM_YOLOV4 (3)  // YOLOV4
#define AILIA_DETECTOR_ALGORITHM_YOLOX (4)   // YOLOX
#define AILIA_DETECTOR_ALGORITHM_SSD (8)     // SSD(Single Shot multibox Detector)

#define AILIA_DETECTOR_FLAG_NORMAL (0)  // オプションなし

/****************************************************************
 * 物体検出API
 **/

int AILIA_API ailiaCreateDetector(struct AILIADetector** detector, struct AILIANetwork* net, unsigned int format, unsigned int channel, unsigned int range, unsigned int algorithm, unsigned int category_count, unsigned int flags);

void AILIA_API ailiaDestroyDetector(struct AILIADetector* detector);

int AILIA_API ailiaDetectorCompute(struct AILIADetector* detector, const void* src, unsigned int src_stride, unsigned int src_width, unsigned int src_height, unsigned int src_format, float threshold, float iou);

int AILIA_API ailiaDetectorGetObjectCount(struct AILIADetector* detector, unsigned int* obj_count);

int AILIA_API ailiaDetectorGetObject(struct AILIADetector* detector, AILIADetectorObject* obj, unsigned int obj_idx, unsigned int version);

int AILIA_API ailiaDetectorSetAnchors(struct AILIADetector* detector, float* anchors, unsigned int anchors_count);

int AILIA_API ailiaDetectorSetInputShape(struct AILIADetector* detector, unsigned int input_width, unsigned int input_height);

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