#ifndef OBJECT_DETECTION_H
#define OBJECT_DETECTION_H

/*********************************
 * 目标检测类，定义了接受检测结果
 * 的容器方便结果数据传输
 * @EVLIS-W
 * date:2019/9/23
 * *******************************/
#include <unistd.h>
#include <iostream>
#include <iomanip>
#include <string>
#include <vector>
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "tengine_c_api.h"
#include <sys/time.h>
#include "cpu_device.h"


class Object_detection
{
    public:
        struct Box
        {
            float x0;
            float y0;
            float x1;
            float y1;
            int class_idx;
            float score;
        };

        std::vector<float> detection;//目标及其坐标信息

        Object_detection();

        //输入图片，以及ssd模型对应的格式大小
        void get_input_data_ssd(cv::Mat img, float* input_data, int img_h, int img_w);

        //输出结果
        void post_process_ssd(cv::Mat img, float threshold, float* outdata, int num);
};

#endif // OBJECT_DETECTION_H