#ifndef __EB3516_NNIE_H__
#define __EB3516_NNIE_H__

#ifdef __cplusplus
extern "C" {
#endif

#include <stdbool.h>
#include "util.h"

typedef struct {
    int cls;     // The category of the object, an integer > 0
    float score; // Object's credibility score
    RectBox box; // The rectangular area of the object (pixels)
} DetectObjInfo;

int nnie_init();
void nnie_exit();
int nnie_start();
int nnie_stop();
void nnie_callback(DetectObjInfo *detections, int count);

#ifdef __cplusplus
}
#endif

#endif