zhuzizyf
2023-04-22 e5e677f28b90199b5a3e04dcdb1c9cfede6bd6de
funasr/runtime/onnxruntime/src/FsmnVad.h
@@ -1,3 +1,4 @@
// Collaborators: zhuzizyf(China Telecom Shanghai)
#ifndef VAD_SERVER_FSMNVAD_H
#define VAD_SERVER_FSMNVAD_H
@@ -16,6 +17,7 @@
                  float vad_speech_noise_thres);
    std::vector<std::vector<int>> infer(const std::vector<float> &waves);
    void Reset();
private:
@@ -35,12 +37,15 @@
            std::vector<std::vector<float>> *out_prob);
    void load_cmvn(const char *filename);
    void init_cache();
    std::shared_ptr<Ort::Session> vad_session_ = nullptr;
    Ort::Env env_;
    Ort::SessionOptions session_options_;
    std::vector<const char *> vad_in_names_;
    std::vector<const char *> vad_out_names_;
    std::vector<std::vector<float>> in_cache_;
    knf::FbankOptions fbank_opts;
    std::vector<float> means_list;
    std::vector<float> vars_list;