#ifndef DNDETECT_H
#define DNDETECT_H
#define GPU
#include <darknet.h>
#include <opencv2/opencv.hpp>
class DnDetect
{
public:
    DnDetect(const int gpuIndex);
    virtual ~DnDetect();

    std::vector<cv::Rect2f> detect(cv::Mat&);
private:
    image matToImg(cv::Mat& RefImg);

private:
    network *m_net;
    float m_thresh;
    float m_hier_thresh;
    float m_nms;
    char **names;

    image **alphabet;
};

#endif