src.cpp
1.05 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
#include "./ai_engine_module/VehicleAnalysis.h"
#include "./ai_engine_module/include.h"
#include "./utils/logger.hpp"
int main() {
VehicleAnalysis va;
va.init(0, 16);
string filename = "./test.jpg";
aclrtStream stream = nullptr;
ACL_CALL(aclrtCreateStream(&stream), ACL_SUCCESS, SY_FAILED);
DvppProcess* dvpp = new DvppProcess();
dvpp->InitResource(stream);
const int batchsize = 1;
vector<ImageData> vec_img;
ImageData dvpp_data;
for (int b = 0; b < batchsize; b++) {
ImageData src;
Utils::ReadImageFile(src, filename); //将二进制图像读入内存,并读取宽高信息
ACL_CALL(dvpp->CvtJpegToYuv420sp(dvpp_data, src), SY_SUCCESS, SY_FAILED); //解码
vec_img.push_back(dvpp_data);
}
printf("1 \n");
va.detect(vec_img);
delete dvpp;
dvpp = nullptr;
if (stream != nullptr) {
int ret = aclrtDestroyStream(stream);
if (ret != ACL_SUCCESS) {
LOG_ERROR("destroy stream failed");
}
stream = nullptr;
}
return 0;
}