speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix code style

Browse Source
This commit is contained in:
lyblsgo 2023-04-20 15:52:11 +08:00
parent ae3e256760
commit 11ae574450
4 changed files with 4 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
funasr/runtime/onnxruntime/src/FsmnVad.cc
View File

@ -1,6 +1,4 @@
//
// Created by root on 4/9/23.
//
#include <fstream>
#include "FsmnVad.h"
#include "precomp.h"
@ -9,8 +7,6 @@
void FsmnVad::init_vad(const std::string &vad_model, const std::string &vad_cmvn, int vad_sample_rate, int vad_silence_duration, int vad_max_len,
float vad_speech_noise_thres) {
session_options_.SetIntraOpNumThreads(1);
session_options_.SetGraphOptimizationLevel(ORT_ENABLE_ALL);
session_options_.DisableCpuMemArena();
@ -33,7 +29,6 @@ void FsmnVad::init_vad(const std::string &vad_model, const std::string &vad_cmvn
}
void FsmnVad::read_model(const std::string &vad_model) {
try {
vad_session_ = std::make_shared<Ort::Session>(
@ -46,8 +41,6 @@ void FsmnVad::read_model(const std::string &vad_model) {
GetInputOutputInfo(vad_session_, &vad_in_names_, &vad_out_names_);
}
void FsmnVad::GetInputOutputInfo(
const std::shared_ptr<Ort::Session> &session,
std::vector<const char *> *in_names, std::vector<const char *> *out_names) {

3
funasr/runtime/onnxruntime/src/FsmnVad.h
View File

@ -1,6 +1,3 @@
//
// Created by zyf on 4/9/23.
//
#ifndef VAD_SERVER_FSMNVAD_H
#define VAD_SERVER_FSMNVAD_H

3
funasr/runtime/onnxruntime/src/e2e_vad.h
View File

@ -1,6 +1,3 @@
//
// Created by root on 3/31/23.
//
#include <utility>
#include <vector>

6
funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
View File

@ -35,12 +35,12 @@ ModelImp::ModelImp(const char* path,int nNumThread, bool quantize, bool use_vad)
fbank_opts.frame_opts.frame_length_ms = 25;
fbank_opts.energy_floor = 0;
fbank_opts.mel_opts.debug_mel = false;
//fbank_ = std::make_unique<knf::OnlineFbank>(fbank_opts);
// fbank_ = std::make_unique<knf::OnlineFbank>(fbank_opts);
//sessionOptions.SetInterOpNumThreads(1);
//sessionOptions.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_EXTENDED);
// sessionOptions.SetInterOpNumThreads(1);
sessionOptions.SetIntraOpNumThreads(nNumThread);
sessionOptions.SetGraphOptimizationLevel(ORT_ENABLE_ALL);
// DisableCpuMemArena can improve performance
sessionOptions.DisableCpuMemArena();
#ifdef _WIN32