File size: 3,072 Bytes
d21d362 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | #include <iostream>
#include <vector>
#include <cmath>
#include <iomanip> // std::fixed, std::setprecision
#include <fstream> // std::ifstream
#include <string> // std::string
// 自定义头文件
#include "wav.h" // 包含 wav::WavReader 定义
#include "time_stamp.h" // 包含 timestamp_t 定义
#include "vad_iterator.h" // 包含 VadIterator 类声明
int main(int argc, char* argv[]) {
if (argc < 3) {
std::cerr << "Usage: " << argv[0] << " <model_absolute_path>" << " <audio_list_absolute_path>" << std::endl;
return 1;
}
// 获取命令行传入的模型路径和音频列表文件路径
std::string model_path = argv[1];
std::string audio_list_path = argv[2];
// 打开 audio_list.txt 文件
std::ifstream audio_list_file(audio_list_path);
if (!audio_list_file.is_open()) {
std::cerr << "Error: Unable to open audio list file: " << audio_list_path << std::endl;
return 1;
}
// 初始化 VadIterator
VadIterator vad(model_path);
// 逐行读取音频文件路径并处理
std::string wav_path;
while (std::getline(audio_list_file, wav_path)) {
if (wav_path.empty()) {
continue; // 跳过空行
}
std::cout << wav_path << std::endl;
try {
// 读取 WAV 文件 (expects 16000 Hz, mono, PCM)
wav::WavReader wav_reader(wav_path);
int numSamples = wav_reader.num_samples();
std::vector<float> input_wav(static_cast<size_t>(numSamples));
for (size_t i = 0; i < static_cast<size_t>(numSamples); i++) {
input_wav[i] = static_cast<float>(*(wav_reader.data() + i));
}
// 处理音频
vad.process(input_wav);
// 获取语音时间戳 (以样本为单位)
std::vector<timestamp_t> stamps = vad.get_speech_timestamps();
// 将时间戳转换为秒并输出
const float sample_rate_float = 16000.0f;
for (size_t i = 0; i < stamps.size(); i++) {
float start_sec = std::rint((stamps[i].start / sample_rate_float) * 10.0f) / 10.0f;
float end_sec = std::rint((stamps[i].end / sample_rate_float) * 10.0f) / 10.0f;
// std::cout << "Speech detected from "
// << std::fixed << std::setprecision(1) << start_sec
// << " s to "
// << std::fixed << std::setprecision(1) << end_sec
// << " s"
// << " [ " << stamps[i].start << " " << stamps[i].end << " ]"
// << std::endl;
std::cout << stamps[i].start << ", " << stamps[i].end << std::endl;
}
// 重置内部状态
vad.reset();
} catch (const std::exception& e) {
std::cerr << "Error processing file " << wav_path << ": " << e.what() << std::endl;
}
}
audio_list_file.close();
return 0;
} |