From 5464448bd3668da9acbb2fe6ee462916f3dae91c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=BE=E8=81=AA?= Date: Sun, 25 Jun 2023 22:57:47 +0800 Subject: [PATCH] update funasrruntime.cpp --- funasr/runtime/onnxruntime/src/funasrruntime.cpp | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/funasr/runtime/onnxruntime/src/funasrruntime.cpp b/funasr/runtime/onnxruntime/src/funasrruntime.cpp index f41c71a1b..69d9e96e6 100644 --- a/funasr/runtime/onnxruntime/src/funasrruntime.cpp +++ b/funasr/runtime/onnxruntime/src/funasrruntime.cpp @@ -51,6 +51,9 @@ extern "C" { int flag = 0; funasr::FUNASR_RECOG_RESULT* p_result = new funasr::FUNASR_RECOG_RESULT; p_result->snippet_time = audio.GetTimeLen(); + if(p_result->snippet_time == 0){ + return p_result; + } int n_step = 0; int n_total = audio.GetQueueSize(); while (audio.Fetch(buff, len, flag) > 0) { @@ -90,6 +93,9 @@ extern "C" { int n_total = audio.GetQueueSize(); funasr::FUNASR_RECOG_RESULT* p_result = new funasr::FUNASR_RECOG_RESULT; p_result->snippet_time = audio.GetTimeLen(); + if(p_result->snippet_time == 0){ + return p_result; + } while (audio.Fetch(buff, len, flag) > 0) { string msg = recog_obj->Forward(buff, len, flag); p_result->msg += msg; @@ -114,6 +120,9 @@ extern "C" { funasr::FUNASR_VAD_RESULT* p_result = new funasr::FUNASR_VAD_RESULT; p_result->snippet_time = audio.GetTimeLen(); + if(p_result->snippet_time == 0){ + return p_result; + } vector> vad_segments; audio.Split(vad_obj, vad_segments, input_finished); @@ -143,6 +152,9 @@ extern "C" { funasr::FUNASR_VAD_RESULT* p_result = new funasr::FUNASR_VAD_RESULT; p_result->snippet_time = audio.GetTimeLen(); + if(p_result->snippet_time == 0){ + return p_result; + } vector> vad_segments; audio.Split(vad_obj, vad_segments, true); @@ -181,6 +193,9 @@ extern "C" { int flag = 0; funasr::FUNASR_RECOG_RESULT* p_result = new funasr::FUNASR_RECOG_RESULT; p_result->snippet_time = audio.GetTimeLen(); + if(p_result->snippet_time == 0){ + return p_result; + } int n_step = 0; int n_total = audio.GetQueueSize(); while (audio.Fetch(buff, len, flag) > 0) {