From a2a307bbc7e72d6631ec47fb77120cf0fafa1306 Mon Sep 17 00:00:00 2001 From: Qianhe Chen <54462604+chenqianhe@users.noreply.github.com> Date: Tue, 31 Jan 2023 22:15:22 +0800 Subject: [PATCH] [Bug Fix] Fix vad example preprocess (#1197) fix preprocess stop sharing --- examples/audio/silero-vad/cpp/vad.cc | 32 +++++++++++----------------- 1 file changed, 13 insertions(+), 19 deletions(-) diff --git a/examples/audio/silero-vad/cpp/vad.cc b/examples/audio/silero-vad/cpp/vad.cc index 5b4516059..512d94adb 100644 --- a/examples/audio/silero-vad/cpp/vad.cc +++ b/examples/audio/silero-vad/cpp/vad.cc @@ -94,25 +94,19 @@ void Vad::setAudioCofig(int sr, int frame_ms, float threshold, } bool Vad::Preprocess(std::vector audioWindowData) { - fastdeploy::FDTensor inputTensor, srTensor, hTensor, cTensor; - inputTensor.SetExternalData(input_node_dims_, fastdeploy::FDDataType::FP32, - audioWindowData.data()); - inputTensor.name = "input"; - srTensor.SetExternalData(sr_node_dims_, fastdeploy::FDDataType::INT64, - sr_.data()); - srTensor.name = "sr"; - hTensor.SetExternalData(hc_node_dims_, fastdeploy::FDDataType::FP32, - _h.data()); - hTensor.name = "h"; - cTensor.SetExternalData(hc_node_dims_, fastdeploy::FDDataType::FP32, - _c.data()); - cTensor.name = "c"; - - inputTensors_.clear(); - inputTensors_.emplace_back(inputTensor); - inputTensors_.emplace_back(srTensor); - inputTensors_.emplace_back(hTensor); - inputTensors_.emplace_back(cTensor); + inputTensors_.resize(4); + inputTensors_[0].name = "input"; + inputTensors_[0].SetExternalData(input_node_dims_, fastdeploy::FDDataType::FP32, + audioWindowData.data()); + inputTensors_[1].name = "sr"; + inputTensors_[1].SetExternalData(sr_node_dims_, fastdeploy::FDDataType::INT64, + sr_.data()); + inputTensors_[2].name = "h"; + inputTensors_[2].SetExternalData(hc_node_dims_, fastdeploy::FDDataType::FP32, + _h.data()); + inputTensors_[3].name = "c"; + inputTensors_[3].SetExternalData(hc_node_dims_, fastdeploy::FDDataType::FP32, + _c.data()); return true; }