1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
| #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
|
|