File size: 2,335 Bytes
139f25e | 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 | // WakeUp.axera wake-word detection example: reads raw int16 16k mono PCM,
// feeds 512-sample frames, prints per-frame score and trigger events.
#include "wake_word.hpp"
#include <cstdlib>
#include <cstdint>
#include <cstdio>
#include <fstream>
#include <iterator>
#include <vector>
int main(int argc, char** argv) {
if (argc < 3) {
std::fprintf(stderr, "usage: %s <model.axmodel> <pcm_s16le_16k.raw> [threshold]\n", argv[0]);
return 1;
}
const float threshold = argc >= 4 ? static_cast<float>(std::atof(argv[3])) : 0.615f;
std::ifstream f(argv[2], std::ios::binary);
if (!f) {
std::fprintf(stderr, "failed to open %s\n", argv[2]);
return 1;
}
std::vector<int16_t> pcm((std::istreambuf_iterator<char>(f)),
std::istreambuf_iterator<char>());
if (pcm.size() % 2) pcm.pop_back();
std::vector<int16_t> samples;
samples.reserve(pcm.size() / 2);
for (size_t i = 0; i < pcm.size(); i += 2) {
samples.push_back(static_cast<int16_t>(pcm[i] | (pcm[i + 1] << 8)));
}
const int n_frames = (static_cast<int>(samples.size()) - 512) / 512 + 1;
if (n_frames < 1) {
std::fprintf(stderr, "PCM too short\n");
return 1;
}
try {
WakeWordDetector detector(argv[1], threshold, 3);
int triggers = 0;
float max_score = -1e9f, max_sum = -1e9f;
int max_frame = 0;
for (int i = 0; i < n_frames; ++i) {
WakeWordDetector::FrameResult r = detector.ProcessFrame(&samples[i * 512]);
if (r.wake_score > max_score) max_score = r.wake_score;
if (r.window_sum > max_sum) {
max_sum = r.window_sum;
max_frame = i;
}
if (r.triggered) {
++triggers;
std::printf("TRIGGER at frame %d (%.2fs) score=%.4f sum=%.4f\n",
i, i * 0.032, r.wake_score, r.window_sum);
}
}
std::printf("frames=%d max_score=%.4f@frame%d(%.2fs) max_sum=%.4f triggers=%d threshold=%.3f\n",
n_frames, max_score, max_frame, max_frame * 0.032, max_sum, triggers, threshold);
return 0;
} catch (const std::exception& exc) {
std::fprintf(stderr, "error: %s\n", exc.what());
return 1;
}
}
|