From 8a100b731efba8c18f7e7b6cb1cb04ded94248b3 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期二, 21 三月 2023 14:52:15 +0800
Subject: [PATCH] add aishell-1 rnnt egs
---
funasr/runtime/onnxruntime/include/Audio.h | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/funasr/runtime/onnxruntime/include/Audio.h b/funasr/runtime/onnxruntime/include/Audio.h
index af8d2a9..da5e82c 100644
--- a/funasr/runtime/onnxruntime/include/Audio.h
+++ b/funasr/runtime/onnxruntime/include/Audio.h
@@ -43,11 +43,17 @@
Audio(int data_type, int size);
~Audio();
void disp();
- bool loadwav(const char *filename);
+ bool loadwav(const char* filename);
+ bool loadwav(const char* buf, int nLen);
+ bool loadpcmwav(const char* buf, int nFileLen);
+ bool loadpcmwav(const char* filename);
int fetch_chunck(float *&dout, int len);
int fetch(float *&dout, int &len, int &flag);
void padding();
void split();
+ float get_time_len();
+
+ int get_queue_size() { return (int)frame_queue.size(); }
};
#endif
--
Gitblit v1.9.1