From 093badf90c29908d75469a1ddb08697e7ad17859 Mon Sep 17 00:00:00 2001 From: AllentDan <41138331+AllentDan@users.noreply.github.com> Date: Mon, 30 Jan 2023 20:50:06 +0800 Subject: [PATCH] fix rknn output index error in SDK (#1683) --- csrc/mmdeploy/net/rknn/rknn_net.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/csrc/mmdeploy/net/rknn/rknn_net.cpp b/csrc/mmdeploy/net/rknn/rknn_net.cpp index 9851b7ba2b..9e3a58a345 100644 --- a/csrc/mmdeploy/net/rknn/rknn_net.cpp +++ b/csrc/mmdeploy/net/rknn/rknn_net.cpp @@ -250,7 +250,7 @@ Result 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(); }