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