ailia_tracker.h

Go to the documentation of this file.

 
#ifndef INCLUDED_AILIA_TRACKER
#define INCLUDED_AILIA_TRACKER
 
/* エラーコードと構造体 */
 
#include "ailia.h"
#include "ailia_detector.h"
 
/* 呼び出し規約 */
 
#if defined(_WIN64) || defined(_M_X64) || defined(__amd64__) || defined(__x86_64__) || defined(__APPLE__) || \
    defined(__ANDROID__) || defined(ANDROID) || defined(__linux__) || defined(NN_NINTENDO_SDK)
#define AILIA_API
#else
#define AILIA_API __stdcall
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
/****************************************************************
 * アルゴリズム定義
 **/
 
#define AILIA_TRACKER_ALGORITHM_BYTE_TRACK (0)
 
/****************************************************************
 * 物体情報
 **/
#define AILIA_TRACKER_OBJECT_VERSION (1)
 
typedef struct _AILIATrackerObject {
    unsigned int id;
    unsigned int category;
    float prob;
    float x;
    float y;
    float w;
    float h;
} AILIATrackerObject;
 
/****************************************************************
 * 設定情報
 **/
#define AILIA_TRACKER_SETTINGS_VERSION (1)
 
typedef struct _AILIATrackerSettings {
    float score_threshold;
    float nms_threshold;
    float track_threshold;
    int track_buffer;
    float match_threshold;
} AILIATrackerSettings;
 
/****************************************************************
 * ネットワークオブジェクトのインスタンス
 **/
 
struct AILIATracker;
 
/****************************************************************
 * Tracking API
 **/
 
int AILIA_API ailiaTrackerCreate(struct AILIATracker** tracker, int algorithm, AILIATrackerSettings settings, int version);
 
int AILIA_API
ailiaTrackerAddTarget(struct AILIATracker* tracker, AILIADetectorObject target_object);
 
int AILIA_API
ailiaTrackerCompute(struct AILIATracker* tracker);
 
int AILIA_API ailiaTrackerGetObjectCount(struct AILIATracker* tracker, unsigned int* obj_count);
 
int AILIA_API ailiaTrackerGetObject(struct AILIATracker* tracker,
                                    AILIATrackerObject* obj,
                                    unsigned int index,
                                    unsigned int version);
 
int AILIA_API ailiaTrackerDestroy(struct AILIATracker *tracker);
 
const char* AILIA_API ailiaTrackerGetErrorDetail(struct AILIATracker* tracker);
 
#ifdef __cplusplus
}
#endif
#endif /* !defined(INCLUDED_AILIA_TRACKER) */