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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
| #ifndef MODEL_H
| #define MODEL_H
| #include "dataType.h"
| #include <algorithm>
| #include "../config.h"
|
| // * Each rect's data structure.
| // * tlwh: topleft point & (w,h).
| // * confidence: detection confidence.
| // * feature: the rect's 128d feature.
| // */
| class DETECTION_ROW
| {
| public:
| int obj_id;
| DETECTBOX tlwh;
| float confidence;
| FEATURE feature;
| DETECTBOX to_xyah() const;
| DETECTBOX to_tlbr() const;
|
| bool isFall;
| int fallScore;
|
| bool isRun;
| int runScore;
|
| bool is_hat;
| int hatScore;
|
| int helmetScore;
|
| int headScore;
|
| bool is_mask;
| int maskScore;
|
| bool is_smoke;
| int smokeScore;
|
| };
|
| typedef std::vector<DETECTION_ROW> DETECTIONS;
|
|
| #endif // MODEL_H
|
|
