fix rknn output index error in SDK (#1682)

open-mmlab · Jan 30, 2023 · 878a364 · 878a364
1 parent cbddf5a
commit 878a364
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/mmdeploy/net/rknn/rknn_net.cpp b/csrc/mmdeploy/net/rknn/rknn_net.cpp
@@ -250,7 +250,7 @@ Result<void> RKNNNet::Forward() {
   for (uint32_t i = 0; i < output_tensors_.size(); ++i) {
     outputs[i].want_float = 1;
     outputs[i].is_prealloc = 1;  // use pre-allocated buffer in `output_tensors_`
-    outputs[i].index = 1;
+    outputs[i].index = i;
     outputs[i].buf = output_tensors_[i].data();
     outputs[i].size = output_tensors_[i].byte_size();
   }