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

Merge pull request #407 from bygreencn/master

Browse Source 
Fix a bug at c sample code and some bugs at wav.h.
This commit is contained in:
Alexander Veysov
2023-12-17 20:26:40 +03:00
committed by GitHub
parent 03dc3fae5c 0b7da6e74b
commit 94504ece54
2 changed files with 53 additions and 28 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;
// default + parameterized constructor
timestamp_t(int start = 0, int end = 0)
timestamp_t(int start = -1, int end = -1)
: start(start), end(end)
{
};
@@ -129,7 +129,7 @@ private:
prev_end = next_start = 0;
speeches.clear();
current_speech = timestamp_t(0, 0);
current_speech = timestamp_t();
};
void predict(const std::vector<float> &data)
@@ -198,7 +198,7 @@ private:
if (prev_end > 0) {
current_speech.end = prev_end;
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)
if (next_start < prev_end)
@@ -214,7 +214,7 @@ private:
else{
current_speech.end = current_sample;
speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0);
current_speech = timestamp_t();
prev_end = 0;
next_start = 0;
temp_end = 0;
@@ -268,7 +268,7 @@ private:
if (current_speech.end - current_speech.start > min_speech_samples)
{
speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0);
current_speech = timestamp_t();
prev_end = 0;
next_start = 0;
temp_end = 0;
@@ -297,10 +297,10 @@ public:
predict(r);
}
if (current_speech.start > 0) {
if (current_speech.start >= 0) {
current_speech.end = audio_length_samples;
speeches.push_back(current_speech);
current_speech = timestamp_t(0, 0);
current_speech = timestamp_t();
prev_end = 0;
next_start = 0;
temp_end = 0;

67
examples/cpp/wav.h
View File

@@ -79,6 +79,13 @@ class WavReader {
fread(header.data, 8, sizeof(char), fp);
}
if (header.data_size == 0) {
int offset = ftell(fp);
fseek(fp, 0, SEEK_END);
header.data_size = ftell(fp) - offset;
fseek(fp, offset, SEEK_SET);
}
num_channel_ = header.channels;
sample_rate_ = header.sample_rate;
bits_per_sample_ = header.bit;
@@ -90,35 +97,53 @@ class WavReader {
std::cout << "sample_rate_ :" << sample_rate_ << std::endl;
std::cout << "bits_per_sample_:" << bits_per_sample_ << std::endl;
std::cout << "num_samples :" << num_data << std::endl;
std::cout << "num_data size :" << header.data_size << std::endl;
std::cout << "num_data_size :" << header.data_size << std::endl;
for (int i = 0; i < num_data; ++i) {
switch (bits_per_sample_) {
switch (bits_per_sample_) {
case 8: {
char sample;
fread(&sample, 1, sizeof(char), fp);
data_[i] = static_cast<float>(sample);
break;
char sample;
for (int i = 0; i < num_data; ++i) {
fread(&sample, 1, sizeof(char), fp);
data_[i] = static_cast<float>(sample) / 32768;
}
break;
}
case 16: {
int16_t sample;
fread(&sample, 1, sizeof(int16_t), fp);
// std::cout << sample;
data_[i] = static_cast<float>(sample);
// std::cout << data_[i];
break;
int16_t sample;
for (int i = 0; i < num_data; ++i) {
fread(&sample, 1, sizeof(int16_t), fp);
data_[i] = static_cast<float>(sample) / 32768;
}
break;
}
case 32: {
int sample;
fread(&sample, 1, sizeof(int), fp);
data_[i] = static_cast<float>(sample);
break;
case 32:
{
if (header.format == 1) //S32
{
int sample;
for (int i = 0; i < num_data; ++i) {
fread(&sample, 1, sizeof(int), fp);
data_[i] = static_cast<float>(sample) / 32768;
}
}
else if (header.format == 3) // IEEE-float
{
float sample;
for (int i = 0; i < num_data; ++i) {
fread(&sample, 1, sizeof(float), fp);
data_[i] = static_cast<float>(sample);
}
}
else {
printf("unsupported quantization bits\n");
}
break;
}
default:
printf("unsupported quantization bits\n");
exit(1);
}
printf("unsupported quantization bits\n");
break;
}
fclose(fp);
return true;
}