雾聪
2024-09-25 3e44172c8b927ffc69b585d4fd80b458cb18ba97
runtime/onnxruntime/src/fsmn-vad.h
@@ -28,10 +28,13 @@
        std::vector<std::vector<float>> *in_cache,
        bool is_final);
    void Reset();
    int GetVadSampleRate() { return vad_sample_rate_; };
    
    std::shared_ptr<Ort::Session> vad_session_ = nullptr;
    Ort::Env env_;
    Ort::SessionOptions session_options_;
    Ort::AllocatorWithDefaultOptions vad_allocator;
    std::vector<const char *> vad_in_names_;
    std::vector<const char *> vad_out_names_;
    std::vector<std::vector<float>> in_cache_;
@@ -54,7 +57,8 @@
    static void GetInputOutputInfo(
            const std::shared_ptr<Ort::Session> &session,
            std::vector<const char *> *in_names, std::vector<const char *> *out_names);
            std::vector<const char *> *in_names, std::vector<const char *> *out_names,
            Ort::AllocatorWithDefaultOptions *allocator);
    void FbankKaldi(float sample_rate, std::vector<std::vector<float>> &vad_feats,
                    std::vector<float> &waves);