| | |
| | | public: |
| | | ParaformerTorch(); |
| | | ~ParaformerTorch(); |
| | | void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, int thread_num); |
| | | void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, int thread_num); |
| | | void InitHwCompiler(const std::string &hw_model, int thread_num); |
| | | void InitSegDict(const std::string &seg_dict_model); |
| | | std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords); |
| | |
| | | string Rescoring(); |
| | | string GetLang(){return language;}; |
| | | int GetAsrSampleRate() { return asr_sample_rate; }; |
| | | void SetBatchSize(int batch_size) {batch_size_ = batch_size}; |
| | | void SetBatchSize(int batch_size) {batch_size_ = batch_size;}; |
| | | int GetBatchSize() {return batch_size_;}; |
| | | void StartUtterance(); |
| | | void EndUtterance(); |