common.h
3.26 KB
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#ifndef COMMON_H_
#define COMMON_H_
#include <functional>
#include "header.h"
#include "opencv2/opencv.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "MyAtomic.hpp"
#include "../SfxDecoder/DxDecoderWrap.h"
#define IMG_CHANNELS 3
#define SCALE_OUT 10 //判断目标框初始位置时，在最小距离的基础上适当外扩
#define MAX_OBJ_COUNT 100
struct OBJ_KEY {
int videoID;
int objID;
bool operator< (OBJ_KEY const& _A) const
{
if (videoID < _A.videoID) return true;
if (videoID == _A.videoID) return objID < _A.objID;
return false;
}
bool operator== (OBJ_KEY const& _A) const
{
if (videoID == _A.videoID && objID == _A.objID)
return true;
else
return false;
}
};
struct FRAME_KEY {
int videoID;
unsigned int objID;
bool operator< (FRAME_KEY const& _A) const
{
if (videoID < _A.videoID) return true;
if (videoID == _A.videoID) return objID < _A.objID;
return false;
}
bool operator== (FRAME_KEY const& _A) const
{
if (videoID == _A.videoID && objID == _A.objID)
return true;
else
return false;
}
};
using SNAPSHOT_CALLBACK = std::function<void(video_object_snapshot *)>;
using REALTIME_CALLBACK = std::function<void(unsigned char *img_data, int img_height, int img_width)>;
using FINISH_CALLBACK = std::function<void( const int )>;
using OBJECT_INFO_CALLBACK = std::function<void(video_object_info *)>;
struct VideoHeightWidth{
double height;
double width;
};
enum TaskState
{
PLAY,
PAUSE,
FINISH,
DECODEERROR //解码线程可能报错，报错之后直接结束掉该路解码
};
struct Task{
int taskID;
const char *taskFileSource;
TaskState taskState;
DxDecoderWrap *taskTcuvid;
DxGPUFrame task_algorithm_data; //����¿�ܲ���resize�Ĵ���������ʱ��backup��ԭͼ��С��ͼƬ�ͽ��㷨�������
bool bInited{false};
VideoHeightWidth taskHeightWidth;
MyAtomic<int> taskFrameCount;
MyAtomic<int> taskLastFrameCount;
int taskTotalFrameCount{1};
SNAPSHOT_CALLBACK taskObjCallbackFunc;
REALTIME_CALLBACK taskRealTimeCallbackFunc;
FINISH_CALLBACK taskFinishCallbackFunc{nullptr};
cv::Mat frameImage;
char* folderNameLittle;
char* folderName;
char* folderNameFace;
sy_rect task_min_boxsize[DETECTTYPE];
unsigned long long timestamp;
};
typedef struct VPT_ObjInfo //����ṹ��
{
int left;
int top;
int right;
int bottom;
int center_x;
int center_y;
int index; // ����/�������֧��10��
long id; // Ŀ��ΨһID��ͬһIDΪͬһĿ��
int num; // ��ID�����µĵ�num֡
double confidence; // ���Ŷ�
int snap_flag;
}VPT_ObjInfo;
typedef struct VPT_Result
{
int objCount {0};
VPT_ObjInfo obj[MAX_OBJ_COUNT];
}VPT_Result;
struct CommandConfig {
sy_command hp_analysis_config;
sy_command hcp_analysis_config;
sy_command vehicle_analysis_config;
sy_command vehicle_recg_config;
sy_command vehicle_plate_det_recg_config;
sy_command hf_recg_config;
sy_command hcf_recg_config;
sy_command vcf_recg_config;
sy_command face_detect_config;
};
#endif