VehiclePlate.cpp
2.73 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
#include "VehiclePlate.h"
#include <cuda_runtime.h>
#include "../../FFNvDecoder/logger.hpp"
VehiclePlate::VehiclePlate(/* args */)
{
}
VehiclePlate::~VehiclePlate()
{
LOG_INFO("~VehiclePlate");
}
int VehiclePlate::init(int gpuid, char* auth_license)
{
vp_detect_param param;
param.gpuid = gpuid;
param.thresld = 0.4;
param.max_batch = 20;
param.serialize_file = "./serialize_file/VPDetect";
param.auth_license = "sy_va_sub_sdk_2023";
vpr_param rparam;
rparam.gpuid = gpuid;
rparam.max_batch = 20;
rparam.serialize_file = "./serialize_file/VPRecg";
param.auth_license = "sy_va_sub_sdk_2023";
int ret = vpdr_init(&handle, param, rparam);
if (ret != 0){
LOG_ERROR("vpdr_init Failed!");
return FAILED;
}
LOG_INFO("vpdr_init success! gpu_id: {}", gpuid);
return SUCCESS;
}
int VehiclePlate::process(sy_img * batch_img, int batchsize, vehicle_plate_result *&result)
{
sy_img cpu_batch_img[batchsize];
for (int i = 0; i < batchsize; i++)
{
if (batch_img[i].data_ == NULL) {
LOG_ERROR("data null ");
return FAILED;
}
cpu_batch_img[i].data_ = (unsigned char *)malloc(batch_img[i].w_ *batch_img[i].h_ * batch_img[i].c_ * sizeof(unsigned char));
cudaMemcpy(cpu_batch_img[i].data_, batch_img[i].data_, batch_img[i].w_ *batch_img[i].h_ * batch_img[i].c_ * sizeof(unsigned char), cudaMemcpyDeviceToHost);
cpu_batch_img[i].w_ = batch_img[i].w_;
cpu_batch_img[i].h_ = batch_img[i].h_;
cpu_batch_img[i].c_ = batch_img[i].c_;
}
vehicle_plate_results * all_result = new vehicle_plate_results[batchsize] {};
for (int i = 0; i<batchsize; i++) {
for (int j = 0; j < MAXPLATECOUNT; j++) {
memset(&all_result[i].vehicle_plate_infos[j].rect, -1, sizeof(sy_rect));
}
}
int index_count = vpdr_batch(handle, cpu_batch_img, batchsize, all_result);
cudaDeviceSynchronize();
for (int b = 0; b < batchsize; b++)
{
double max_score = 0.0;
int best_index = 0;
for (int j = 0; j < MAXPLATECOUNT; j++)
{
if (all_result[b].vehicle_plate_infos[j].rect.left_ == -1)
{
continue;
}
if (all_result[b].vehicle_plate_infos[j].detect_score > max_score)
{
max_score = all_result[b].vehicle_plate_infos[j].detect_score;
best_index = j;
}
}
memcpy(&result[b], &all_result[b].vehicle_plate_infos[best_index], sizeof(vehicle_plate_result));
}
for (int i = 0; i < batchsize; i++) {
if(cpu_batch_img[i].data_ != nullptr) {
free(cpu_batch_img[i].data_);
cpu_batch_img[i].data_ = nullptr;
}
}
if (all_result != NULL)
delete [] all_result;
return SUCCESS;
}
int VehiclePlate::release()
{
if (handle) {
vpdr_release(&handle);
handle = nullptr;
}
LOG_INFO("release");
return SUCCESS;
}