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(); }