| | |
|---|
| | | #include <map> |
|---|
| | | #include <vector> |
|---|
| | | |
|---|
| | | namespace funasr { |
|---|
| | | class VadModel { |
|---|
| | | public: |
|---|
| | | virtual ~VadModel(){}; |
|---|
| | | virtual void InitVad(const std::string &vad_model, const std::string &vad_cmvn, const std::string &vad_config, int thread_num)=0; |
|---|
| | | virtual std::vector<std::vector<int>> Infer(const std::vector<float> &waves)=0; |
|---|
| | | virtual void ReadModel(const char* vad_model)=0; |
|---|
| | | virtual void LoadConfigFromYaml(const char* filename)=0; |
|---|
| | | virtual void FbankKaldi(float sample_rate, std::vector<std::vector<float>> &vad_feats, |
|---|
| | | const std::vector<float> &waves)=0; |
|---|
| | | virtual std::vector<std::vector<float>> &LfrCmvn(std::vector<std::vector<float>> &vad_feats)=0; |
|---|
| | | virtual void Forward( |
|---|
| | | const std::vector<std::vector<float>> &chunk_feats, |
|---|
| | | std::vector<std::vector<float>> *out_prob)=0; |
|---|
| | | virtual void LoadCmvn(const char *filename)=0; |
|---|
| | | virtual void InitCache()=0; |
|---|
| | | virtual std::vector<std::vector<int>> Infer(std::vector<float> &waves, bool input_finished=true)=0; |
|---|
| | | }; |
|---|
| | | |
|---|
| | | VadModel *CreateVadModel(std::map<std::string, std::string>& model_path, int thread_num); |
|---|
| | | VadModel *CreateVadModel(void* fsmnvad_handle); |
|---|
| | | } // namespace funasr |
|---|
| | | #endif |
|---|
