输入一个batch数据，RKNN输出是对应的吗？（已解决）

问一下论坛里的各路大神及技术支持，官方说1.2版本RKNN支持输入一个batch，刚好我也有需要，所以写了一下：

1. // Start Inference
   
2.   rknn_input inputs[2];
   
3.   rknn_output outputs[4];
   
4.   rknn_tensor_attr outputs_attr[4];
   
5. 
   
6.   int ret = 0;
   
7.   rknn_context ctx = 0;
   
8. 
   
9.   ret = rknn_init(&ctx, pmodel, model_len, RKNN_FLAG_PRIOR_MEDIUM);
   
10.   if(ret < 0) {
    
11.     printf("rknn_init fail! ret=%d\n", ret);
    
12.     return;
    
13.   }
    
14. 
    
15.   for(int i = 0;i < 4;i++){
    
16.       outputs_attr[i].index = i;
    
17.       ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(outputs_attr[i]), sizeof(outputs_attr[i]));
    
18.       if(ret < 0) {
    
19.           printf("rknn_query fail! ret=%d\n", ret);
    
20.           return;
    
21.       }
    
22.   }
    
23. 
    
24.   if (thread_id > sizeof(multi_npu_process_initialized) / sizeof(int) - 1){
    
25.       return;
    
26.   }
    
27.   multi_npu_process_initialized[thread_id] = 1;
    
28.   printf("The initialization of NPU Process %d has been completed.\n", thread_id);
    
29. 
    
30.   pair<Mat, Mat> pairIndexImage;
    
31.   while (true) {
    
32.       mtxQueueInput.lock();
    
33.       if (queueInput.size() < 2) {
    
34.           mtxQueueInput.unlock();
    
35.           usleep(20000);
    
36.       } else {
    
37.               // Get an image from input queue
    
38.           pairIndexImage = queueInput.front();  // the same time only allow one pthread operation to change the pairIndexImage .
    
39.           queueInput.pop();
    
40.           mtxQueueInput.unlock();
    
41.       }
    
42. 
    
43.       cv::resize(pairIndexImage.first, image_1, cv::Size(img_width, img_height), (0, 0), (0, 0), cv::INTER_LINEAR);
    
44.       cv::resize(pairIndexImage.second, image_2, cv::Size(img_width, img_height), (0, 0), (0, 0), cv::INTER_LINEAR);
    
45. 
    
46.       inputs[0].index = input_index;
    
47.       inputs[0].buf = image_1.data;
    
48.       inputs[0].size = img_width * img_height * img_channels;
    
49.       inputs[0].pass_through = false;
    
50.       inputs[0].type = RKNN_TENSOR_UINT8;
    
51.       inputs[0].fmt = RKNN_TENSOR_NHWC;
    
52. 
    
53.       inputs[1].index = input_index;
    
54.       inputs[1].buf = image_2.data;
    
55.       inputs[1].size = img_width * img_height * img_channels;
    
56.       inputs[1].pass_through = false;
    
57.       inputs[1].type = RKNN_TENSOR_UINT8;
    
58.       inputs[1].fmt = RKNN_TENSOR_NHWC;
    
59.       ret = rknn_inputs_set(ctx, 2, inputs);
    
60.       if(ret < 0) {
    
61.           printf("rknn_input_set fail! ret=%d\n", ret);
    
62.           return;
    
63.       }
    
64. 
    
65.       ret = rknn_run(ctx, nullptr);
    
66.       if(ret < 0) {
    
67.           printf("rknn_run fail! ret=%d\n", ret);
    
68.           return;
    
69.       }
    
70. 
    
71.       outputs[0].want_float = true;
    
72.       outputs[0].is_prealloc = false;
    
73.       outputs[1].want_float = true;
    
74.       outputs[1].is_prealloc = false;
    
75. 
    
76.       outputs[2].want_float = true;
    
77.       outputs[2].is_prealloc = false;
    
78.       outputs[3].want_float = true;
    
79.       outputs[3].is_prealloc = false;
    
80. 
    
81.       ret = rknn_outputs_get(ctx, 4, outputs, nullptr);
    
82.       if(ret < 0) {
    
83.           printf("rknn_outputs_get fail! ret=%d\n", ret);
    
84.           return;
    
85.       }

复制代码关键是RKNN的输出吧，请问 inputs[0] 是与 outpus[0]、outpus[1]   ，  inputs[1] 是与 outpus[2]、outpus[3]  对应的吗？有需要我可以贴完整代码，方便各位帮忙。

理论上是对应的，你是发现了不对应的情况么？

还没来得及试，npu驱动版本不匹配，先问一下再说。

请问C/C++接口的教程或示例在哪里呀，我只看到python的接口

yuys 发表于 2019-10-24 10:30
请问C/C++接口的教程或示例在哪里呀，我只看到python的接口

你是说算法C++版本？如果是的话，你买板子，板子里没sample demo的吗？http://t.rock-chips.com/forum.php?mod=viewthread&tid=349  sample demo