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

Fix a bug in c code sample when only one timestamp and start from 0.

Browse Source
This commit is contained in:
bygreencn
2023-12-17 22:47:37 +08:00
parent 03dc3fae5c
commit efb5effc8f
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/cpp/silero-vad-onnx.cpp
View File

@@ -26,7 +26,7 @@ public:
int end; int end;
// default + parameterized constructor // default + parameterized constructor
timestamp_t(int start = 0, int end = 0) timestamp_t(int start = -1, int end = -1)
: start(start), end(end) : start(start), end(end)
{ {
}; };
@@ -129,7 +129,7 @@ private:
prev_end = next_start = 0; prev_end = next_start = 0;
speeches.clear(); speeches.clear();
current_speech = timestamp_t(0, 0); current_speech = timestamp_t();
}; };
void predict(const std::vector<float> &data) void predict(const std::vector<float> &data)
@@ -198,7 +198,7 @@ private:
if (prev_end > 0) { if (prev_end > 0) {
current_speech.end = prev_end; current_speech.end = prev_end;
speeches.push_back(current_speech); speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0); current_speech = timestamp_t();
// previously reached silence(< neg_thres) and is still not speech(< thres) // previously reached silence(< neg_thres) and is still not speech(< thres)
if (next_start < prev_end) if (next_start < prev_end)
@@ -214,7 +214,7 @@ private:
else{ else{
current_speech.end = current_sample; current_speech.end = current_sample;
speeches.push_back(current_speech); speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0); current_speech = timestamp_t();
prev_end = 0; prev_end = 0;
next_start = 0; next_start = 0;
temp_end = 0; temp_end = 0;
@@ -268,7 +268,7 @@ private:
if (current_speech.end - current_speech.start > min_speech_samples) if (current_speech.end - current_speech.start > min_speech_samples)
{ {
speeches.push_back(current_speech); speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0); current_speech = timestamp_t();
prev_end = 0; prev_end = 0;
next_start = 0; next_start = 0;
temp_end = 0; temp_end = 0;
@@ -297,10 +297,10 @@ public:
predict(r); predict(r);
} }
if (current_speech.start > 0) { if (current_speech.start >= 0) {
current_speech.end = audio_length_samples; current_speech.end = audio_length_samples;
speeches.push_back(current_speech); speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0); current_speech = timestamp_t();
prev_end = 0; prev_end = 0;
next_start = 0; next_start = 0;
temp_end = 0; temp_end = 0;