e8beee4d
Hu Chunming
实现jni接口
|
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
|
#include "com_objecteye_nativeinterface_vehicle_VehicleNativeInterface.h"
#include "sy_common.h"
#include "sy_errorinfo.h"
#include "iostream"
#include "map"
#include <string>
#include <string.h>
#include "../src/village_pic_interface.h"
#include "opencv2/imgcodecs/legacy/constants_c.h"
#include "opencv2/imgproc/types_c.h"
#include <opencv2/opencv.hpp>
#include "base64.h"
#include "curl/curl.h"
//dvpp����
#include "dvpp_process.h"
#include "utils.h"
//#include "img_codec.h"
double msecond() {
struct timeval tv;
gettimeofday(&tv, 0);
return (tv.tv_sec * 1000.0 + tv.tv_usec / 1000.0);
}
string Decode(const char* data, int size)
{
const char DecodeTable[] =
{
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
62, // '+'
0, 0, 0,
63, // '/'
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, // '0'-'9'
0, 0, 0, 0, 0, 0, 0,
0, 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, // 'A'-'Z'
0, 0, 0, 0, 0, 0,
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, // 'a'-'z'
};
int outSize = 0;
int nValue;
int i= 0;
string outStr = "";
while (i < size)
{
if (*data != '\r' && *data != '\n')
{
nValue = DecodeTable[*data++] << 18;
nValue += DecodeTable[*data++] << 12;
outStr += (nValue & 0x00FF0000) >> 16;
outSize++;
if (*data != '=')
{
nValue += DecodeTable[*data++] << 6;
outStr += (nValue & 0x0000FF00) >> 8;
outSize++;
if (*data != '=')
{
nValue += DecodeTable[*data++];
outStr += nValue & 0x000000FF;
outSize++;
}
}
i += 4;
}
else
{
data++;
i++;
}
}
return outStr;
}
using namespace std;
#define SUCCESS 0 //�ɹ�
#define FAILED -1 //ʧ��
/**
* ��������
* */
sy_format getImgByteType(int type);
size_t write_data(char *ptr, size_t size, size_t nmemb, void *userdata);
//function to retrieve the image as cv::Mat data type
cv::Mat curlImg(const char *img_url, int timeout);
/**
* ȫ�ֱ���
* */
sy_command global_vehicle_detect_config; //�Ƿ����������(һ�������)
sy_command global_vehicle_recg_config; //�Ƿ���������ʶ��
sy_command global_vehicle_recg_supplement_config; //�Ƿ���������ʶ��ʶ��(��ͷ��βlogo���ʶ��+����15����)
sy_command global_vehicle_plate_det_recg_config; //�Ƿ��������Ƽ��ʶ��
sy_command global_vehicle_color_config; //�Ƿ�����������ɫʶ��
sy_command global_vehicle_pendant_det_config; //�Ƿ�������������ʶ��
sy_command global_vehicle_illegal_config; //�Ƿ�������Υ����Ϊ���
sy_command global_vehicle_feature_config; //�Ƿ���������������ȡ
//sy_command global_vehicle_special_config; //�Ƿ���������Ʒ���ʶ��
//sy_command global_vehicle_image_quality_config; //�Ƿ���������ͼ��������ʶ��
sy_command global_vehicle_motor_tricycle_analysis_config; //�Ƿ�����Ħ�г��Ƿ�����/Ħ�г���ʻ���Ƿ��ͷ��/ũ�ó��Ƿ�����
//sy_command global_vehicle_motor_hs_output_config; //�Ƿ�����Ħ�г�δ��ͷ�����ͷ������
//sy_command global_vehicle_stain_vp_config; //�Ƿ������ڵ�����ʶ��
//sy_command global_vehicle_muck_truck_cover_config; //�Ƿ������������Ǹ�ʶ��(ǰ�����vehicle_special_config)
sy_command global_vehicle_manned_config; //�Ƿ������������ֳ�����ʶ��
//sy_command global_vehicle_pose_config; //�Ƿ�����pose��� //20210618 atlas
int gpu_id;
/*
* Class: com_objecteye_nativeinterface_vehicle_VehicleNativeInterface
* Method: va_init
* Signature: ([JLcom/objecteye/pojo/vehicle/VehicleAnalysisParam;)I
*/
JNIEXPORT jint JNICALL Java_com_objecteye_nativeinterface_vehicle_VehicleNativeInterface_init(JNIEnv *env, jobject obj, jlongArray handle, jobject vehicleAnalysisParam)
{
//getFieldID
//�㷨���ò���
jclass cls_vehcileAnalysisParam = env->GetObjectClass(vehicleAnalysisParam);
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
129
|
jfieldID fid_sdk_path = env->GetFieldID(cls_vehcileAnalysisParam, "sdk_path", "Ljava/lang/String;");
|
e8beee4d
Hu Chunming
实现jni接口
|
130
131
|
jfieldID fid_gpuid = env->GetFieldID(cls_vehcileAnalysisParam, "gpuId", "I");
|
e3062370
Hu Chunming
优化日志
|
132
133
134
135
136
|
jfieldID fid_aiEngineParam_logLevel = env->GetFieldID(cls_vehcileAnalysisParam, "logLevel", "I");
jfieldID fid_aiEngineParam_logPath = env->GetFieldID(cls_vehcileAnalysisParam, "logPath", "Ljava/lang/String;");
jfieldID fid_aiEngineParam_logDays = env->GetFieldID(cls_vehcileAnalysisParam, "logDays", "I");
jfieldID fid_aiEngineParam_logMem = env->GetFieldID(cls_vehcileAnalysisParam, "logMem", "D");
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
137
138
|
jstring str_sdk_path = (jstring)env->GetObjectField(vehicleAnalysisParam, fid_sdk_path);
const char *sdk_path = env->GetStringUTFChars(str_sdk_path, JNI_FALSE);
|
e8beee4d
Hu Chunming
实现jni接口
|
139
140
|
jint gpuid = env->GetIntField(vehicleAnalysisParam, fid_gpuid);
|
e3062370
Hu Chunming
优化日志
|
141
142
143
144
145
146
147
|
//log
jstring str_aiEngineParam_logPath = (jstring)env->GetObjectField(vehicleAnalysisParam, fid_aiEngineParam_logPath);
const char* aiEngineParam_logPath = env->GetStringUTFChars(str_aiEngineParam_logPath, JNI_FALSE);
jint aiEngineParam_logDays = env->GetIntField(vehicleAnalysisParam, fid_aiEngineParam_logDays);
jdouble aiEngineParam_logMem = env->GetDoubleField(vehicleAnalysisParam, fid_aiEngineParam_logMem);
jint aiEngineParam_logLevel = env->GetIntField(vehicleAnalysisParam, fid_aiEngineParam_logLevel);
|
e8beee4d
Hu Chunming
实现jni接口
|
148
149
150
|
void *vaHandle = NULL;
VillageParam param;
param.dev_id = gpuid;
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
151
|
param.sdk_path = sdk_path;
|
e3062370
Hu Chunming
优化日志
|
152
153
154
155
156
157
158
|
param.log_level = aiEngineParam_logLevel;
param.log_path = aiEngineParam_logPath;
param.log_days = aiEngineParam_logDays;
param.log_mem = aiEngineParam_logMem;
printf("jni dev_id:%d sdk_path:%s_log_level:%d\n", param.dev_id, param.sdk_path.c_str(), param.log_level);
|
e8beee4d
Hu Chunming
实现jni接口
|
159
|
int ret = village_pic_init(&vaHandle, param);
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
160
|
if (ret != SUCCESS) {
|
e8beee4d
Hu Chunming
实现jni接口
|
161
162
163
|
printf("jni info:va_init failed.");
return ret;
}
|
e8beee4d
Hu Chunming
实现jni接口
|
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
|
if (ret == SUCCESS)
{
global_vehicle_detect_config = SY_CONFIG_OPEN;
global_vehicle_recg_config = SY_CONFIG_OPEN;
global_vehicle_recg_supplement_config = SY_CONFIG_OPEN;
global_vehicle_plate_det_recg_config = SY_CONFIG_OPEN;
global_vehicle_pendant_det_config = SY_CONFIG_OPEN;
global_vehicle_motor_tricycle_analysis_config = SY_CONFIG_OPEN;
global_vehicle_manned_config = SY_CONFIG_OPEN;
global_vehicle_illegal_config = SY_CONFIG_CLOSE;
global_vehicle_feature_config = SY_CONFIG_CLOSE;
global_vehicle_color_config = SY_CONFIG_CLOSE;
jlong temp[1];
temp[0] = (jlong)vaHandle;
env->SetLongArrayRegion(handle, 0, 1, temp);
}
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
182
|
env->ReleaseStringUTFChars(str_sdk_path, sdk_path);
|
e8beee4d
Hu Chunming
实现jni接口
|
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
|
return ret;
}
void CvBGR2NV21(cv::Mat& bgr, unsigned char* yuv) {
int stride = (bgr.cols + 127) / 128 * 128;
int strideH = (bgr.rows + 15) / 16 * 16;
for (int i = 0; i < bgr.rows; i++) {
for (int j = 0; j < bgr.cols; j++) {
int B = bgr.at<cv::Vec3b>(i, j)[0];
int G = bgr.at<cv::Vec3b>(i, j)[1];
int R = bgr.at<cv::Vec3b>(i, j)[2];
int Y = (77 * R + 150 * G + 29 * B) >> 8;
yuv[i * stride + j] = (Y < 0) ? 0 : ((Y > 255) ? 255 : Y);
if (i % 2 == 0 && j % 2 == 0) {
int U = ((-44 * R - 87 * G + 131 * B) >> 8) + 128;
int V = ((131 * R - 110 * G - 21 * B) >> 8) + 128;
yuv[strideH * stride + i / 2 * stride + j] = (V < 0) ? 0 : ((V > 255) ? 255 : V);
yuv[strideH * stride + i / 2 * stride + j + 1] = (U < 0) ? 0 : ((U > 255) ? 255 : U);
}
}
}
}
/*
* Class: com_objecteye_nativeinterface_vehicle_VehicleNativeInterface
* Method: batch
* Signature: (J[Lcom/objecteye/pojo/common/SyImgParam;II[Lcom/objecteye/pojo/vehicle/VehicleAnalysisResultParam;)I
*/
JNIEXPORT jint JNICALL Java_com_objecteye_nativeinterface_vehicle_VehicleNativeInterface_batch(JNIEnv *env, jobject jobj, jlong handle, jobjectArray syImgParams, jint format, jint batchSize, jobjectArray vehicleAnalysisResultParams)
{
int ret = FAILED;
void *vaHandle = (void *)handle;
// ��ȡͼƬ����
jclass cls_syImgParam = env->FindClass("com/objecteye/pojo/common/SyImgParam");
jfieldID fid_image_str = env->GetFieldID(cls_syImgParam, "image_string", "Ljava/lang/String;");//base64�����ⲿͳһ��url��fileת��base64����
sy_img batch_img[batchSize];
cv::Mat car_images[batchSize];
ImageData dvpp_data[batchSize];
aclrtContext ctx;
ACL_CALL(aclrtCreateContext(&ctx, gpu_id), ACL_ERROR_NONE, FAILED);
ACL_CALL(aclrtSetCurrentContext(ctx), ACL_ERROR_NONE, FAILED);
//��ʼ��dvpp
aclrtStream stream = nullptr;
ACL_CALL(aclrtCreateStream(&stream), ACL_ERROR_NONE, FAILED);
DvppProcess* dvpp = new DvppProcess();
dvpp->InitResource(stream);
//Base64Utils decoder();
//wh20210623
std::shared_ptr<uint8_t> NV21data[batchSize];
if (format != 0 && format != 1)
{
printf("ERROR FORMAT TYPE: %d\n", format);
return -1;
}
double t1,t2,t3;
t1 = msecond();
for (size_t i = 0; i < batchSize; i++)
{
jobject syImgParam = env->GetObjectArrayElement(syImgParams, i);
jstring imageString1 = (jstring)env->GetObjectField(syImgParam, fid_image_str);
const char *image_str = env->GetStringUTFChars(imageString1, JNI_FALSE);
string imageString = image_str;
//printf("format = %d\n", format);
if (format == 0) //url
{
printf("url is not surposed \n");
}
else
{
string decode_str;
//decode_str = decoder->Decode(encode_str[b].data(), encode_str[b].size());
decode_str = Decode(imageString.data(), imageString.size());
//debug========================================================
ImageData src;
uint32_t binFileBufferLen = decode_str.size() + 8;//�ֽ���
uint8_t* binFileBufferData = new(std::nothrow) uint8_t[binFileBufferLen];
if (binFileBufferData == nullptr) {
ERROR_LOG("malloc binFileBufferData failed");
return FAILED;
}
memcpy((char *)binFileBufferData, decode_str.data(), binFileBufferLen);
int32_t ch = 0;
ACL_CALL(acldvppJpegGetImageInfo(binFileBufferData, binFileBufferLen,
&(src.width), &(src.height), &ch), SUCCESS, FAILED);
src.data.reset(binFileBufferData, [](uint8_t* p) { delete[](p); });
src.size = binFileBufferLen;
// printf("vpdr_api_batch:img.w=%d ,img.h=%d \n",src.width,src.height);
ACL_CALL(dvpp->CvtJpegToYuv420sp(dvpp_data[i], src), SUCCESS, FAILED);
batch_img[i].w_ = dvpp_data[i].width;//dvpp_data[b].alignWidth;
batch_img[i].h_ = dvpp_data[i].height;//dvpp_data[b].alignHeight;
//batch_img[i].data_ = (uint8_t*)Utils::CopyDataDeviceToLocal(dvpp_data[i].data.get(), dvpp_data[i].size);//dvpp_data[b].data.get();
batch_img[i].data_ = dvpp_data[i].data.get();
}
env->ReleaseStringUTFChars(imageString1, image_str);
}
t2 = msecond();
//printf("va jni info:decode time: %.2f\n", (t2 - t1));
|
8d319775
Hu Chunming
添加prob字段
|
305
|
std::cout << "village_pic_analysis " << std::endl;
|
427201b2
Hu Chunming
完善结果返回
|
306
307
308
|
std::vector<AnalysisResult> vec_result = village_pic_analysis(vaHandle, batch_img, batchSize);
if (vec_result.size() <= 0)
|
e8beee4d
Hu Chunming
实现jni接口
|
309
310
311
312
313
314
315
316
317
318
319
320
321
322
|
{
printf("jni info:village_pic_analysis failed.");
return -1;
}
//cout << "VAJNI_TEST va_1batch -> end ...." << endl;
t3 = msecond();
//printf("va jni info:process time: %.2f\n", (t3 - t2));
for (int b = 0; b < batchSize; b++)
{
//printf("VAJNI_TEST begin reult: %d, count: %d\n", b, result[b].count);
|
427201b2
Hu Chunming
完善结果返回
|
323
|
for (int c = 0; c < vec_result[b].info.size(); c++)
|
e8beee4d
Hu Chunming
实现jni接口
|
324
325
326
327
|
{
//std::cout << "obj_quality_idx=" << result[b].info[c].obj_quality_idx << std::endl; //聚档数据类型�?-无效数据�?-低质数据�?-优质数据
std::string str_vehicle_type;
|
427201b2
Hu Chunming
完善结果返回
|
328
|
int vehicle_type = vec_result[b].info[c].type;
|
e8beee4d
Hu Chunming
实现jni接口
|
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
|
//std::cout << "vehicle_type=" << vehicle_type << std::endl;
if (vehicle_type == 0)str_vehicle_type = "head";
else if (vehicle_type == 1)str_vehicle_type = "rear";
else if (vehicle_type == 2)str_vehicle_type = "motor";
else if (vehicle_type == 3)str_vehicle_type = "tricycle";
else if (vehicle_type == 4)str_vehicle_type = "body";
//std::cout << "vehicle_type info:" << str_vehicle_type << std::endl;
}
}
if (ret == SUCCESS)
{
// getClass
jclass cls_vehicleAnalysisResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleAnalysisResultParam");
jclass cls_VehicleInfo = env->FindClass("com/objecteye/pojo/vehicle/VehicleInfoParam");
|
cee436d5
Hu Chunming
完成道路信息返回
|
345
346
347
|
jclass cls_LineInfo = env->FindClass("com/objecteye/pojo/vehicle/LineInfoParam");
jclass cls_SegInfo = env->FindClass("com/objecteye/pojo/vehicle/SegInfoParam");
|
e8beee4d
Hu Chunming
实现jni接口
|
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
|
jclass cls_VehicleDetectResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleDetectResultParam");
jclass cls_VehicleColorResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleColorResultParam");
jclass cls_VehicleRecgResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleRecognizeResultParam");
//jclass cls_VehicleRecgTopnResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleRecognizeResultTopnParam");//wh20210527������topn�ӿ�
jclass cls_VplateResult = env->FindClass("com/objecteye/pojo/vehicle/VehiclePlateResultParam");
jclass cls_VehiclePendantDetResult = env->FindClass("com/objecteye/pojo/vehicle/VehiclePendantDetectResultParam");
jclass cls_VehicleIllegalDetResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleIllegalDetectResultParam");
jclass cls_VehicleFeaResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleFeatureResultParam");
jclass cls_SyRect = env->FindClass("com/objecteye/pojo/common/SyRectParam");
jclass cls_VplateNum = env->FindClass("com/objecteye/pojo/vehicle/VehiclePlateNumParam");
jclass cls_VehiclePendantDetInfo = env->FindClass("com/objecteye/pojo/vehicle/VehiclePendantDetectInfoParam");
jclass cls_VehicleIllegalDetInfo = env->FindClass("com/objecteye/pojo/vehicle/VehicleIllegalDetectInfoParam");
jclass cls_VehicleIllegalDetDetails = env->FindClass("com/objecteye/pojo/vehicle/VehicleIllegalDetectDetailsParam");
//jclass cls_VehicleSpecialInformationParam = env->FindClass("com/objecteye/pojo/vehicle/VehicleSpecialInformationParam");
//jclass cls_VehicleSpecialResult = env->FindClass("com/objecteye/pojo/vehicle/VehicleSpecialResultParam");
//jclass cls_MucktruckcoverResult = env->FindClass("com/objecteye/pojo/vehicle/MucktruckcoverResultParam");
jclass cls_MtaResult = env->FindClass("com/objecteye/pojo/vehicle/MtaResultParam");
jclass cls_MtaDetailsParam = env->FindClass("com/objecteye/pojo/vehicle/MtaDetailsParam");
//jclass cls_StainVplateResultParam = env->FindClass("com/objecteye/pojo/vehicle/StainVplateResultParam");
jclass cls_MannedResultParam = env->FindClass("com/objecteye/pojo/vehicle/MannedResultParam");
|
cee436d5
Hu Chunming
完成道路信息返回
|
368
|
|
e8beee4d
Hu Chunming
实现jni接口
|
369
370
|
//jclass cls_VehiclePoseResultParam = env->FindClass("com/objecteye/pojo/vehicle/VehiclePoseResultParam");//wh20210601������pose���
|
cee436d5
Hu Chunming
完成道路信息返回
|
371
|
jclass cls_SyPoint = env->FindClass("com/objecteye/pojo/common/SyPointParam");
|
e8beee4d
Hu Chunming
实现jni接口
|
372
373
374
|
//getmethodId
|
cee436d5
Hu Chunming
完成道路信息返回
|
375
|
jmethodID mid_vehicleAnalysisResult = env->GetMethodID(cls_vehicleAnalysisResult, "<init>", "([Lcom/objecteye/pojo/vehicle/VehicleInfoParam;com/objecteye/pojo/vehicle/LineInfoParam;com/objecteye/pojo/vehicle/SegInfoParam;I)V");
|
e8beee4d
Hu Chunming
实现jni接口
|
376
|
|
8d319775
Hu Chunming
添加prob字段
|
377
|
jmethodID mid_VehicleInfo = env->GetMethodID(cls_VehicleInfo, "<init>", "(Lcom/objecteye/pojo/vehicle/VehicleDetectResultParam;Lcom/objecteye/pojo/vehicle/VehicleDetectResultParam;Lcom/objecteye/pojo/vehicle/VehicleDetectResultParam;Lcom/objecteye/pojo/vehicle/VehicleColorResultParam;Lcom/objecteye/pojo/vehicle/VehicleRecognizeResultParam;Lcom/objecteye/pojo/vehicle/VehiclePlateResultParam;Lcom/objecteye/pojo/vehicle/VehiclePendantDetectResultParam;Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectResultParam;Lcom/objecteye/pojo/vehicle/VehicleFeatureResultParam;Lcom/objecteye/pojo/vehicle/MtaResultParam;Lcom/objecteye/pojo/vehicle/MannedResultParam;I;I;I;F;I;I;I;I;I;I;F;I;I)V");
|
e8beee4d
Hu Chunming
实现jni接口
|
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
|
jmethodID mid_VehicleDetectResult = env->GetMethodID(cls_VehicleDetectResult, "<init>", "(Lcom/objecteye/pojo/common/SyRectParam;F)V");
jmethodID mid_VehicleColorResult = env->GetMethodID(cls_VehicleColorResult, "<init>", "(FI)V");
//jmethodID mid_VehicleRecgTopnResult = env->GetMethodID(cls_VehicleRecgTopnResult, "<init>", "([Lcom/objecteye/pojo/vehicle/VehicleRecognizeResultParam;I)V");//wh20210527������topn�ӿ�
jmethodID mid_VehicleRecgResult = env->GetMethodID(cls_VehicleRecgResult, "<init>", "(Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;F)V");
jmethodID mid_VplateResult = env->GetMethodID(cls_VplateResult, "<init>", "(Lcom/objecteye/pojo/common/SyRectParam;F[Lcom/objecteye/pojo/vehicle/VehiclePlateNumParam;FII)V");
jmethodID mid_VehiclePendantDetResult = env->GetMethodID(cls_VehiclePendantDetResult, "<init>", "([Lcom/objecteye/pojo/vehicle/VehiclePendantDetectInfoParam;I)V");
jmethodID mid_VehicleIllegalDetResult = env->GetMethodID(cls_VehicleIllegalDetResult, "<init>", "(Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectInfoParam;Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectInfoParam;)V");
jmethodID mid_VehicleFeaResult = env->GetMethodID(cls_VehicleFeaResult, "<init>", "([F)V");
jmethodID mid_SyRect = env->GetMethodID(cls_SyRect, "<init>", "(IIII)V");
jmethodID mid_VplateNum = env->GetMethodID(cls_VplateNum, "<init>", "(Ljava/lang/String;F)V");
//jmethodID mid_VehiclePendantDetInfo = env->GetMethodID(cls_VehiclePendantDetInfo, "<init>", "(Lcom/objecteye/pojo/common/SyRectParam;IFI[F)V");
//wh20210916�����
jmethodID mid_VehiclePendantDetInfo = env->GetMethodID(cls_VehiclePendantDetInfo, "<init>", "(Lcom/objecteye/pojo/common/SyRectParam;IFI)V");
jmethodID mid_VehicleIllegalDetInfo = env->GetMethodID(cls_VehicleIllegalDetInfo, "<init>", "(Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectDetailsParam;Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectDetailsParam;Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectDetailsParam;Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectDetailsParam;Lcom/objecteye/pojo/vehicle/VehicleIllegalDetectDetailsParam;)V");
jmethodID mid_VehicleIllegalDetDetails = env->GetMethodID(cls_VehicleIllegalDetDetails, "<init>", "(IF)V");
//jmethodID mid_VehicleSpecialInformationParam = env->GetMethodID(cls_VehicleSpecialInformationParam, "<init>", "(FI)V");
//jmethodID mid_VehicleSpecialResult = env->GetMethodID(cls_VehicleSpecialResult, "<init>", "([Lcom/objecteye/pojo/vehicle/VehicleSpecialInformationParam;I)V");
jmethodID mid_MtaResult = env->GetMethodID(cls_MtaResult, "<init>", "(Lcom/objecteye/pojo/vehicle/MtaDetailsParam;Lcom/objecteye/pojo/vehicle/MtaDetailsParam;Lcom/objecteye/pojo/vehicle/MtaDetailsParam;Lcom/objecteye/pojo/vehicle/VehicleDetectResultParam;)V");
jmethodID mid_MtaDetailsParam = env->GetMethodID(cls_MtaDetailsParam, "<init>", "(IF)V");
//jmethodID mid_StainVplateResultParam = env->GetMethodID(cls_StainVplateResultParam, "<init>", "(FI)V");
//jmethodID mid_MucktruckcoverResult = env->GetMethodID(cls_MucktruckcoverResult, "<init>", "(IF)V");
jmethodID mid_MannedResult = env->GetMethodID(cls_MannedResultParam, "<init>", "([Lcom/objecteye/pojo/vehicle/VehicleDetectResultParam;I)V");//wh20220222
|
cee436d5
Hu Chunming
完成道路信息返回
|
401
|
jmethodID mid_SyPoint = env->GetMethodID(cls_SyPoint, "<init>", "(II)V");//wh20210601������pose���
|
e8beee4d
Hu Chunming
实现jni接口
|
402
|
//jmethodID mid_VehiclePoseResult = env->GetMethodID(cls_VehiclePoseResultParam, "<init>", "([Lcom/objecteye/pojo/common/SyPointParam;[F[FDDD)V");
|
cee436d5
Hu Chunming
完成道路信息返回
|
403
404
405
406
|
jmethodID mid_LineInfo = env->GetMethodID(cls_LineInfo, "<init>", "(Lcom/objecteye/pojo/vehicle/SyPointParam;I)V");
jmethodID mid_SegInfo = env->GetMethodID(cls_SegInfo, "<init>", "(Lcom/objecteye/pojo/vehicle/SyPointParam;I)V");
|
e8beee4d
Hu Chunming
实现jni接口
|
407
|
//set value to param
|
8d319775
Hu Chunming
添加prob字段
|
408
|
std::cout << "vehicleInfoArray " << std::endl;
|
e8beee4d
Hu Chunming
实现jni接口
|
409
410
411
|
for (size_t i = 0; i < batchSize; i++)
{
|
427201b2
Hu Chunming
完善结果返回
|
412
|
jobjectArray vehicleInfoArray = env->NewObjectArray(vec_result[i].info.size(), cls_VehicleInfo, NULL);
|
427201b2
Hu Chunming
完善结果返回
|
413
|
for (int j = 0; j < vec_result[i].info.size(); j++)
|
e8beee4d
Hu Chunming
实现jni接口
|
414
|
{ //vehicleDetectRes
|
427201b2
Hu Chunming
完善结果返回
|
415
416
417
|
auto vehicle_info = vec_result[i].info[j];
jobject vehicle_detect_res_syRect_rect = env->NewObject(cls_SyRect, mid_SyRect, vehicle_info.vehicle_detect_res.rect.left_, vehicle_info.vehicle_detect_res.rect.top_, vehicle_info.vehicle_detect_res.rect.width_, vehicle_info.vehicle_detect_res.rect.height_);
jobject vehicleDetectResult = env->NewObject(cls_VehicleDetectResult, mid_VehicleDetectResult, vehicle_detect_res_syRect_rect, vehicle_info.vehicle_detect_res.score);
|
e8beee4d
Hu Chunming
实现jni接口
|
418
|
//vehicleWinDetectRes
|
427201b2
Hu Chunming
完善结果返回
|
419
420
|
jobject vehicle_win_detect_res_syRect_rect = env->NewObject(cls_SyRect, mid_SyRect, vehicle_info.vehicle_win_detect_res.rect.left_, vehicle_info.vehicle_win_detect_res.rect.top_, vehicle_info.vehicle_win_detect_res.rect.width_, vehicle_info.vehicle_win_detect_res.rect.height_);
jobject vehicleWinDetectResult = env->NewObject(cls_VehicleDetectResult, mid_VehicleDetectResult, vehicle_win_detect_res_syRect_rect, vehicle_info.vehicle_win_detect_res.score);
|
e8beee4d
Hu Chunming
实现jni接口
|
421
|
//vehicleBodyDetectRes
|
427201b2
Hu Chunming
完善结果返回
|
422
423
|
jobject vehicle_body_detect_res_syRect_rect = env->NewObject(cls_SyRect, mid_SyRect, vehicle_info.vehicle_body_detect_res.rect.left_, vehicle_info.vehicle_body_detect_res.rect.top_, vehicle_info.vehicle_body_detect_res.rect.width_, vehicle_info.vehicle_body_detect_res.rect.height_);
jobject vehicleBodyDetectResult = env->NewObject(cls_VehicleDetectResult, mid_VehicleDetectResult, vehicle_body_detect_res_syRect_rect, vehicle_info.vehicle_body_detect_res.score);
|
e8beee4d
Hu Chunming
实现jni接口
|
424
425
426
427
|
//vehicleColorRes
jobject vehicleColorResult = NULL;
if (global_vehicle_color_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
428
|
vehicleColorResult = env->NewObject(cls_VehicleColorResult, mid_VehicleColorResult, vehicle_info.vehicle_color_res.score, vehicle_info.vehicle_color_res.index);
|
e8beee4d
Hu Chunming
实现jni接口
|
429
|
}
|
427201b2
Hu Chunming
完善结果返回
|
430
431
|
//vehicleRecgRes //0618 atlas�Ļ�top1�Ľӿ�
|
e8beee4d
Hu Chunming
实现jni接口
|
432
433
434
|
jobject vehicleRecgResult = NULL;
if (global_vehicle_recg_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
435
436
437
438
439
440
441
|
jstring vehicleBrand = env->NewStringUTF(vehicle_info.vehicle_recg_res.vehicle_brand);
jstring vehicleSubbrand = env->NewStringUTF(vehicle_info.vehicle_recg_res.vehicle_subbrand);
jstring vehicleIssueYear = env->NewStringUTF(vehicle_info.vehicle_recg_res.vehicle_issue_year);
jstring vehicleType = env->NewStringUTF(vehicle_info.vehicle_recg_res.vehicle_type);
jstring freightTon = env->NewStringUTF(vehicle_info.vehicle_recg_res.freight_ton);
vehicleRecgResult = env->NewObject(cls_VehicleRecgResult, mid_VehicleRecgResult, vehicleBrand, vehicleSubbrand, vehicleIssueYear, vehicleType, freightTon, vehicle_info.vehicle_recg_res.name_score);
}
|
e8beee4d
Hu Chunming
实现jni接口
|
442
443
444
445
446
447
448
|
//vehiclePlateDetRecgRes
jobject vplateResult = NULL;
jobject vehicle_plate_det_recg_res_syRect_rect = NULL;
jobjectArray vplateNumArray = NULL;
if (global_vehicle_plate_det_recg_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
449
|
vehicle_plate_det_recg_res_syRect_rect = env->NewObject(cls_SyRect, mid_SyRect, vehicle_info.vehicle_plate_det_recg_res.rect.left_, vehicle_info.vehicle_plate_det_recg_res.rect.top_, vehicle_info.vehicle_plate_det_recg_res.rect.width_, vehicle_info.vehicle_plate_det_recg_res.rect.height_);
|
e8beee4d
Hu Chunming
实现jni接口
|
450
451
452
|
vplateNumArray = env->NewObjectArray(PLATENUM, cls_VplateNum, NULL);
for (size_t k = 0; k < PLATENUM; k++)
{
|
427201b2
Hu Chunming
完善结果返回
|
453
454
|
jstring character = env->NewStringUTF(vehicle_info.vehicle_plate_det_recg_res.recg[k].character);
jobject vplateNum = env->NewObject(cls_VplateNum, mid_VplateNum, character, vehicle_info.vehicle_plate_det_recg_res.recg[k].maxprob);
|
e8beee4d
Hu Chunming
实现jni接口
|
455
456
457
|
env->SetObjectArrayElement(vplateNumArray, k, vplateNum);
}
}
|
427201b2
Hu Chunming
完善结果返回
|
458
459
460
461
|
vplateResult = env->NewObject(cls_VplateResult, mid_VplateResult, vehicle_plate_det_recg_res_syRect_rect, vehicle_info.vehicle_plate_det_recg_res.detect_score, vplateNumArray, vehicle_info.vehicle_plate_det_recg_res.num_score, vehicle_info.vehicle_plate_det_recg_res.type, vehicle_info.vehicle_plate_det_recg_res.special_type);
//vehiclePendantDetRes
|
e8beee4d
Hu Chunming
实现jni接口
|
462
463
464
|
jobject vehiclePendantDetResult = NULL;
if (global_vehicle_pendant_det_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
465
466
|
jobjectArray vehiclePendantDetInfoArray = env->NewObjectArray(vehicle_info.vehicle_pendant_det_res.size(), cls_VehiclePendantDetInfo, NULL);
for (size_t m = 0; m < vehicle_info.vehicle_pendant_det_res.size(); m++)
|
e8beee4d
Hu Chunming
实现jni接口
|
467
|
{
|
427201b2
Hu Chunming
完善结果返回
|
468
469
470
|
auto one_pendant_res = vehicle_info.vehicle_pendant_det_res[m];
jobject vehicle_pendant_det_res_vpd_res_syRect_rect = env->NewObject(cls_SyRect, mid_SyRect, one_pendant_res.rect.left_,
one_pendant_res.rect.top_, one_pendant_res.rect.width_, one_pendant_res.rect.height_);
|
e8beee4d
Hu Chunming
实现jni接口
|
471
|
|
5e69f8fc
Hu Chunming
补充字段;
|
472
|
jobject vehiclePendantDetInfo = env->NewObject(cls_VehiclePendantDetInfo, mid_VehiclePendantDetInfo, vehicle_pendant_det_res_vpd_res_syRect_rect, one_pendant_res.index, one_pendant_res.confidence, one_pendant_res.driver_copilot_info, one_pendant_res.iColor);
|
e8beee4d
Hu Chunming
实现jni接口
|
473
|
|
427201b2
Hu Chunming
完善结果返回
|
474
|
env->SetObjectArrayElement(vehiclePendantDetInfoArray, m, vehiclePendantDetInfo);
|
e8beee4d
Hu Chunming
实现jni接口
|
475
|
}
|
427201b2
Hu Chunming
完善结果返回
|
476
|
vehiclePendantDetResult = env->NewObject(cls_VehiclePendantDetResult, mid_VehiclePendantDetResult, vehiclePendantDetInfoArray, vehicle_info.vehicle_pendant_det_res.size());
|
e8beee4d
Hu Chunming
实现jni接口
|
477
|
}
|
427201b2
Hu Chunming
完善结果返回
|
478
|
|
e8beee4d
Hu Chunming
实现jni接口
|
479
480
481
482
|
//vehicleIllegalDetRes
jobject vehicleIllegalDetResult = NULL;
if (global_vehicle_illegal_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
483
484
485
486
487
488
489
490
491
492
493
494
|
jobject vehicleIllegalDetDetails_driver_belt = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.driver.belt.status, vehicle_info.vehicle_illegal_det_res.driver.belt.confidence);
jobject vehicleIllegalDetDetails_driver_smoke = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.driver.smoke.status, vehicle_info.vehicle_illegal_det_res.driver.smoke.confidence);
jobject vehicleIllegalDetDetails_driver_call = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.driver.call.status, vehicle_info.vehicle_illegal_det_res.driver.call.confidence);
jobject vehicleIllegalDetDetails_driver_phone = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.driver.phone.status, vehicle_info.vehicle_illegal_det_res.driver.phone.confidence);
jobject vehicleIllegalDetDetails_driver_person = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.driver.person.status, vehicle_info.vehicle_illegal_det_res.driver.person.confidence);
jobject vehicleIllegalDetDetails_copilot_belt = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.copilot.belt.status, vehicle_info.vehicle_illegal_det_res.copilot.belt.confidence);
jobject vehicleIllegalDetDetails_copilot_smoke = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.copilot.smoke.status, vehicle_info.vehicle_illegal_det_res.copilot.smoke.confidence);
jobject vehicleIllegalDetDetails_copilot_call = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.copilot.call.status, vehicle_info.vehicle_illegal_det_res.copilot.call.confidence);
jobject vehicleIllegalDetDetails_copilot_phone = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.copilot.phone.status, vehicle_info.vehicle_illegal_det_res.copilot.phone.confidence);
jobject vehicleIllegalDetDetails_copilot_person = env->NewObject(cls_VehicleIllegalDetDetails, mid_VehicleIllegalDetDetails, vehicle_info.vehicle_illegal_det_res.copilot.person.status, vehicle_info.vehicle_illegal_det_res.copilot.person.confidence);
jobject vehicleIllegalDetInfo_driver = env->NewObject(cls_VehicleIllegalDetInfo, mid_VehicleIllegalDetInfo, vehicleIllegalDetDetails_driver_belt, vehicleIllegalDetDetails_driver_smoke, vehicleIllegalDetDetails_driver_phone, vehicleIllegalDetDetails_driver_call, vehicleIllegalDetDetails_driver_person);
|
e8beee4d
Hu Chunming
实现jni接口
|
495
496
497
|
jobject vehicleIllegalDetInfo_copilot = env->NewObject(cls_VehicleIllegalDetInfo, mid_VehicleIllegalDetInfo, vehicleIllegalDetDetails_copilot_belt, vehicleIllegalDetDetails_copilot_smoke, vehicleIllegalDetDetails_copilot_phone, vehicleIllegalDetDetails_copilot_call, vehicleIllegalDetDetails_copilot_person);
vehicleIllegalDetResult = env->NewObject(cls_VehicleIllegalDetResult, mid_VehicleIllegalDetResult, vehicleIllegalDetInfo_driver, vehicleIllegalDetInfo_copilot);
}
|
427201b2
Hu Chunming
完善结果返回
|
498
|
|
e8beee4d
Hu Chunming
实现jni接口
|
499
500
501
502
503
|
//vehicleFeaRes
jobject vehicleFeaResult = NULL;
if (global_vehicle_feature_config == SY_CONFIG_OPEN)
{
jfloatArray feature = env->NewFloatArray(VA_FEATURESIZE);
|
427201b2
Hu Chunming
完善结果返回
|
504
|
env->SetFloatArrayRegion(feature, 0, VA_FEATURESIZE, vehicle_info.vehicle_fea_res.feature);
|
e8beee4d
Hu Chunming
实现jni接口
|
505
506
|
vehicleFeaResult = env->NewObject(cls_VehicleFeaResult, mid_VehicleFeaResult, feature);
}
|
427201b2
Hu Chunming
完善结果返回
|
507
|
|
e8beee4d
Hu Chunming
实现jni接口
|
508
509
510
511
|
//MtaResultParam
jobject mtaResult = NULL;
if (global_vehicle_motor_tricycle_analysis_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
512
513
514
|
jobject mtaDetailsParam_motor_driver_helmeted = env->NewObject(cls_MtaDetailsParam, mid_MtaDetailsParam, vehicle_info.mta_res.motor_driver_helmeted.status, vehicle_info.mta_res.motor_driver_helmeted.confidence);
jobject mtaDetailsParam_motor_manned = env->NewObject(cls_MtaDetailsParam, mid_MtaDetailsParam, vehicle_info.mta_res.motor_manned.status, vehicle_info.mta_res.motor_manned.confidence);
jobject mtaDetailsParam_tricycle_manned = env->NewObject(cls_MtaDetailsParam, mid_MtaDetailsParam, vehicle_info.mta_res.tricycle_manned.status, vehicle_info.mta_res.tricycle_manned.confidence);
|
e8beee4d
Hu Chunming
实现jni接口
|
515
516
517
|
jobject mtaDetailsParam_motor_driver_hs_rect = NULL;
//if (global_vehicle_motor_hs_output_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
518
519
|
jobject mtaDetailsParam_motor_driver_hs_rect_syRect_rect = env->NewObject(cls_SyRect, mid_SyRect, vehicle_info.mta_res.motor_driver_hs_rect.rect.left_, vehicle_info.mta_res.motor_driver_hs_rect.rect.top_, vehicle_info.mta_res.motor_driver_hs_rect.rect.width_, vehicle_info.mta_res.motor_driver_hs_rect.rect.height_);
mtaDetailsParam_motor_driver_hs_rect = env->NewObject(cls_VehicleDetectResult, mid_VehicleDetectResult, mtaDetailsParam_motor_driver_hs_rect_syRect_rect, vehicle_info.mta_res.motor_driver_hs_rect.score);
|
e8beee4d
Hu Chunming
实现jni接口
|
520
521
522
|
}
mtaResult = env->NewObject(cls_MtaResult, mid_MtaResult, mtaDetailsParam_motor_driver_helmeted, mtaDetailsParam_motor_manned, mtaDetailsParam_tricycle_manned, mtaDetailsParam_motor_driver_hs_rect);
}
|
427201b2
Hu Chunming
完善结果返回
|
523
524
|
//LorryMannedResultParam
|
e8beee4d
Hu Chunming
实现jni接口
|
525
526
527
|
jobject MannedResult = NULL;
if (global_vehicle_manned_config == SY_CONFIG_OPEN)
{
|
427201b2
Hu Chunming
完善结果返回
|
528
|
int MannedResHsCount = vehicle_info.manned_res.hs_count;
|
e8beee4d
Hu Chunming
实现jni接口
|
529
530
|
jobjectArray hsDetectResultArray = env->NewObjectArray(MannedResHsCount, cls_VehicleDetectResult, NULL);
|
427201b2
Hu Chunming
完善结果返回
|
531
|
for (size_t m = 0; m < MannedResHsCount; m++)
|
e8beee4d
Hu Chunming
实现jni接口
|
532
533
|
{
jobject Manned_hsDetect_syRect = env->NewObject(cls_SyRect, mid_SyRect,
|
427201b2
Hu Chunming
完善结果返回
|
534
535
536
537
|
vehicle_info.manned_res.hs_rect[m].rect.left_,
vehicle_info.manned_res.hs_rect[m].rect.top_,
vehicle_info.manned_res.hs_rect[m].rect.width_,
vehicle_info.manned_res.hs_rect[m].rect.height_);
|
e8beee4d
Hu Chunming
实现jni接口
|
538
|
|
427201b2
Hu Chunming
完善结果返回
|
539
|
jobject Manned_hsDetect = env->NewObject(cls_VehicleDetectResult, mid_VehicleDetectResult, Manned_hsDetect_syRect, vehicle_info.manned_res.hs_rect[m].score);
|
e8beee4d
Hu Chunming
实现jni接口
|
540
541
542
543
|
env->SetObjectArrayElement(hsDetectResultArray, m, Manned_hsDetect);
}
|
427201b2
Hu Chunming
完善结果返回
|
544
545
546
|
MannedResult = env->NewObject(cls_MannedResultParam, mid_MannedResult,
hsDetectResultArray,
MannedResHsCount);
|
e8beee4d
Hu Chunming
实现jni接口
|
547
548
|
}
|
427201b2
Hu Chunming
完善结果返回
|
549
|
jint type = vehicle_info.type;
|
5e69f8fc
Hu Chunming
补充字段;
|
550
551
|
jint vpt_type = vehicle_info.vpt_type;
jint rainshed = vehicle_info.rainshed; //是否安装雨棚, 0 有雨棚 1 无雨棚
|
8d319775
Hu Chunming
添加prob字段
|
552
|
jfloat rainshed_prob = vehicle_info.rainshed_prob;
|
5e69f8fc
Hu Chunming
补充字段;
|
553
554
555
556
557
558
|
jint truck_manned = vehicle_info.truck_manned;
jint motor_manned = vehicle_info.motor_manned;
jint human_upper_color = vehicle_info.human_upper_color;
jint human_lower_color = vehicle_info.human_lower_color;
jint reverse_driving = vehicle_info.reverse_driving; // 0 正常行驶 1 逆行 -1 未知
jint phoning = vehicle_info.phoning; // 骑车打电话
|
8d319775
Hu Chunming
添加prob字段
|
559
|
jfloat phoning_prob = vehicle_info.phoning_prob;
|
5e69f8fc
Hu Chunming
补充字段;
|
560
561
562
|
jint cross_line = vehicle_info.cross_line; // 压实线, 1 黄实线 2 白实线 -1 其他
jint cross_diversion_line = vehicle_info.cross_diversion_line; // 压导流线 1 压了 -1 未压
|
8d319775
Hu Chunming
添加prob字段
|
563
|
jobject vehicleInfo = env->NewObject(cls_VehicleInfo, mid_VehicleInfo, vehicleDetectResult, vehicleWinDetectResult, vehicleBodyDetectResult, vehicleColorResult, vehicleRecgResult, vplateResult, vehiclePendantDetResult, vehicleIllegalDetResult, vehicleFeaResult,mtaResult, MannedResult, type, vpt_type, rainshed, rainshed_prob, truck_manned, motor_manned, human_upper_color, reverse_driving, phoning, phoning_prob, cross_line, cross_diversion_line);
|
e8beee4d
Hu Chunming
实现jni接口
|
564
|
|
427201b2
Hu Chunming
完善结果返回
|
565
|
env->SetObjectArrayElement(vehicleInfoArray, j, vehicleInfo);
|
e8beee4d
Hu Chunming
实现jni接口
|
566
567
|
}
|
8d319775
Hu Chunming
添加prob字段
|
568
569
|
std::cout << "line info " << std::endl;
|
cee436d5
Hu Chunming
完成道路信息返回
|
570
571
572
573
574
575
576
577
578
|
// 道路标志线
jobjectArray LineInfoArray = env->NewObjectArray(vec_result[i].vec_line.size(), cls_LineInfo, NULL);
for (int j = 0; j < vec_result[i].vec_line.size(); j++)
{
auto line_info = vec_result[i].vec_line[j];
jobjectArray ptArray = env->NewObjectArray(line_info.vec_pt.size(), cls_LineInfo, NULL);
for (int k = 0; k < line_info.vec_pt.size(); k++)
{
auto pt = line_info.vec_pt[k];
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
579
|
jobject line_info_point = env->NewObject(cls_SyPoint, mid_SyPoint, pt.x, pt.y);
|
cee436d5
Hu Chunming
完成道路信息返回
|
580
581
582
583
584
585
586
587
|
env->SetObjectArrayElement(ptArray, k, line_info_point);
}
jobject one_line_info = env->NewObject(cls_LineInfo, mid_LineInfo, ptArray, line_info.line_type);
env->SetObjectArrayElement(LineInfoArray, j, one_line_info);
}
|
8d319775
Hu Chunming
添加prob字段
|
588
589
|
std::cout << "road info " << std::endl;
|
cee436d5
Hu Chunming
完成道路信息返回
|
590
591
592
593
594
595
596
597
598
|
// 道路
jobjectArray RoadInfoArray = env->NewObjectArray(vec_result[i].vec_road.size(), cls_SegInfo, NULL);
for (int j = 0; j < vec_result[i].vec_road.size(); j++)
{
auto road_info = vec_result[i].vec_road[j];
jobjectArray ptArray = env->NewObjectArray(road_info.vec_pt.size(), cls_SyPoint, NULL);
for (int k = 0; k < road_info.vec_pt.size(); k++)
{
auto pt = road_info.vec_pt[k];
|
581a68a4
Hu Chunming
修正parse_road无返回值导...
|
599
|
jobject road_info_point = env->NewObject(cls_SyPoint, mid_SyPoint, pt.x, pt.y);
|
cee436d5
Hu Chunming
完成道路信息返回
|
600
601
602
603
604
605
606
607
608
|
env->SetObjectArrayElement(ptArray, k, road_info_point);
}
jobject one_road_info = env->NewObject(cls_SegInfo, mid_SegInfo, ptArray, road_info.seg_type);
env->SetObjectArrayElement(RoadInfoArray, j, one_road_info);
}
jobject vehicleAnalysisResult = env->NewObject(cls_vehicleAnalysisResult, mid_vehicleAnalysisResult, vehicleInfoArray, LineInfoArray, RoadInfoArray, vec_result[i].info.size());
|
e8beee4d
Hu Chunming
实现jni接口
|
609
610
611
612
613
|
env->SetObjectArrayElement(vehicleAnalysisResultParams, i, vehicleAnalysisResult);
}
}
|
e8beee4d
Hu Chunming
实现jni接口
|
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
|
ACL_CALL(aclrtSetCurrentContext(ctx), ACL_ERROR_NONE, FAILED);
delete dvpp;
dvpp = nullptr;
(void) aclrtDestroyStream(stream);
aclrtDestroyContext(ctx);
return ret;
}
/*
* Class: com_objecteye_nativeinterface_vehicle_VehicleNativeInterface
* Method: va_release
* Signature: (J)V
*/
JNIEXPORT void JNICALL Java_com_objecteye_nativeinterface_vehicle_VehicleNativeInterface_release(JNIEnv *env, jobject obj, jlong handle)
{
void *vaHandle = (void *)handle;
if (vaHandle != NULL)
{
|
427201b2
Hu Chunming
完善结果返回
|
633
|
village_pic_release(&vaHandle);
|
e8beee4d
Hu Chunming
实现jni接口
|
634
635
636
637
638
|
}
}
/*
* Class: com_objecteye_nativeinterface_vehicle_VehicleNativeInterface
|
5e69f8fc
Hu Chunming
补充字段;
|
639
|
* Method: get_village_pic_version
|
e8beee4d
Hu Chunming
实现jni接口
|
640
641
642
643
|
* Signature: ()Ljava/lang/String;
*/
JNIEXPORT jstring JNICALL Java_com_objecteye_nativeinterface_vehicle_VehicleNativeInterface_getVersion(JNIEnv *env, jobject obj)
{
|
5e69f8fc
Hu Chunming
补充字段;
|
644
|
const char *version = get_village_pic_version();
|
e8beee4d
Hu Chunming
实现jni接口
|
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
|
return env->NewStringUTF(version);
}
/**
* ����ͼƬ�ı����ʽ
* */
sy_format getImgByteType(int type)
{
sy_format syFormat;
switch (type)
{
case 0:
syFormat = SY_FORMAT_BGR888;
break;
case 1:
syFormat = SY_FORMAT_BGRA888;
break;
case 2:
syFormat = SY_FORMAT_GRAY8;
break;
case 3:
syFormat = SY_FORMAT_YUV4420P;
break;
case 4:
syFormat = SY_FORMAT_NV12;
break;
case 5:
syFormat = SY_FORMAT_NV21;
break;
default:
syFormat = SY_FORMAT_BGR888;
break;
}
return syFormat;
}
//curl writefunction to be passed as a parameter
// we can't ever expect to get the whole image in one piece,
// every router / hub is entitled to fragment it into parts
// (like 1-8k at a time),
// so insert the part at the end of our stream.
size_t write_data(char *ptr, size_t size, size_t nmemb, void *userdata)
{
vector<uchar> *stream = (vector<uchar>*)userdata;
size_t count = size * nmemb;
stream->insert(stream->end(), ptr, ptr + count);
return count;
}
//function to retrieve the image as cv::Mat data type
cv::Mat curlImg(const char *img_url, int timeout = 10)
{
vector<uchar> stream;
CURL *curl = curl_easy_init();
curl_easy_setopt(curl, CURLOPT_URL, img_url); //the img url
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); // pass the writefunction
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &stream); // pass the stream ptr to the writefunction
curl_easy_setopt(curl, CURLOPT_TIMEOUT, timeout); // timeout if curl_easy hangs,
CURLcode res = curl_easy_perform(curl); // start curl
curl_easy_cleanup(curl); // cleanup
return cv::imdecode(stream, -1); // 'keep-as-is'
}
|