deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-05 01:49:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Changed some source.

Browse Source
This commit is contained in:
Nathan Lee
2024-11-22 06:21:49 +00:00
parent 05e380c1de
commit 0189ebd8af
5 changed files with 43 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/cpp_libtorch/silero_torch.h
View File

@@ -2,7 +2,6 @@
//Created On : 2024-11-18
//Description : silero 5.1 system for torch-script(c++).
//Version : 1.0
//Contact : junghan4242@gmail.com
#ifndef SILERO_TORCH_H
#define SILERO_TORCH_H
@@ -27,11 +26,6 @@ namespace silero{
int end;
};
struct Interval {
float start;
float end;
};
class VadIterator{
public:
@@ -42,10 +36,12 @@ namespace silero{
void SpeechProbs(std::vector<float>& input_wav);
std::vector<silero::Interval> GetSpeechTimestamps();
std::vector<silero::SpeechSegment> GetSpeechTimestamps();
void SetVariables();
float threshold;
int sample_rate;
int window_size_ms;
int min_speech_duration_ms;
int max_duration_merge_ms;
bool print_as_samples;
@@ -70,8 +66,8 @@ namespace silero{
void init_engine(int window_size_ms);
void init_torch_model(const std::string& model_path);
void reset_states();
std::vector<Interval> DoVad();
std::vector<Interval> mergeSpeeches(const std::vector<Interval>& speeches, int duration_merge_samples);
std::vector<SpeechSegment> DoVad();
std::vector<SpeechSegment> mergeSpeeches(const std::vector<SpeechSegment>& speeches, int duration_merge_samples);
};