FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2023-07-21 c542eacb0aadcbc49c63db40429fca4e08f807a4 
 funasr/runtime/onnxruntime/include/audio.h


@@ -11,6 +11,7 @@


#endif




using namespace std;


namespace funasr {




class AudioFrame {


  private:


@@ -32,8 +33,9 @@




class Audio {


  private:


    float *speech_data;


    int16_t *speech_buff;


    float *speech_data=nullptr;


    int16_t *speech_buff=nullptr;


    char* speech_char=nullptr;


    int speech_len;


    int speech_align_len;


    int offset;


@@ -46,18 +48,26 @@


    Audio(int data_type, int size);


    ~Audio();


    void Disp();


    bool LoadWav(const char* filename, int32_t* sampling_rate);


    void WavResample(int32_t sampling_rate, const float *waveform, int32_t n);


    bool LoadWav(const char* buf, int n_len, int32_t* sampling_rate);


    bool LoadWav(const char* filename, int32_t* sampling_rate);


    bool LoadWav2Char(const char* filename, int32_t* sampling_rate);


    bool LoadPcmwav(const char* buf, int n_file_len, int32_t* sampling_rate);


    bool LoadPcmwav(const char* filename, int32_t* sampling_rate);


    bool LoadPcmwav2Char(const char* filename, int32_t* sampling_rate);


    bool LoadOthers2Char(const char* filename);


    bool FfmpegLoad(const char *filename);


    bool FfmpegLoad(const char* buf, int n_file_len);


    int FetchChunck(float *&dout, int len);


    int Fetch(float *&dout, int &len, int &flag);


    void Padding();


    void Split(OfflineStream* offline_streamj);


    void Split(VadModel* vad_obj, vector<std::vector<int>>& vad_segments);


    void Split(VadModel* vad_obj, vector<std::vector<int>>& vad_segments, bool input_finished=true);


    float GetTimeLen();


    int GetQueueSize() { return (int)frame_queue.size(); }


    char* GetSpeechChar(){return speech_char;}


    int GetSpeechLen(){return speech_len;}


};




} // namespace funasr


#endif