Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
45 lines
2.4 KiB
C++
45 lines
2.4 KiB
C++
#include "engine/models/firered_vad/runtime.h"
|
|
|
|
#include <algorithm>
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
|
|
using namespace engine::models::firered_vad;
|
|
|
|
void check(const std::vector<float> & probabilities, const DetectionOptions & options,
|
|
const std::vector<std::pair<int64_t, int64_t>> & expected) {
|
|
const int64_t samples = probabilities.empty() ? 0 : (probabilities.size() - 1) * 160 + 400;
|
|
auto actual = segment_probabilities(probabilities, samples, options);
|
|
if (actual.size() != expected.size()) throw std::runtime_error("VAD segment count differs from upstream fixture");
|
|
for (size_t i = 0; i < actual.size(); ++i)
|
|
if (actual[i].span.start_sample != expected[i].first || actual[i].span.end_sample != expected[i].second)
|
|
throw std::runtime_error("VAD timestamps differ from upstream fixture: segment=" + std::to_string(i) +
|
|
" actual=" + std::to_string(actual[i].span.start_sample) + "," + std::to_string(actual[i].span.end_sample) +
|
|
" expected=" + std::to_string(expected[i].first) + "," + std::to_string(expected[i].second));
|
|
}
|
|
|
|
int main() {
|
|
try {
|
|
DetectionOptions defaults;
|
|
std::vector<float> speech(110, 0.0f);
|
|
std::fill(speech.begin() + 40, speech.begin() + 80, 1.0f);
|
|
check(speech, defaults, {{5760, 16480}});
|
|
std::vector<float> short_speech(20, 0.0f);
|
|
std::fill(short_speech.begin() + 10, short_speech.begin() + 15, 1.0f);
|
|
check(short_speech, defaults, {});
|
|
check({}, defaults, {});
|
|
check(std::vector<float>(100, 0.0f), defaults, {});
|
|
DetectionOptions split;
|
|
split.smooth_window_frames = 3; split.min_speech_frames = 3;
|
|
split.max_speech_frames = 10; split.min_silence_frames = 2;
|
|
check(std::vector<float>(50, 1.0f), split,
|
|
{{0,800},{960,1760},{1920,2720},{2880,3680},{3840,4640},{4800,5600},{5760,6560},{6720,8240}});
|
|
split.max_speech_frames = 100; split.merge_silence_frames = 15; split.extend_speech_frames = 2;
|
|
std::vector<float> merge(50, 0.0f);
|
|
std::fill(merge.begin() + 5, merge.begin() + 15, 1.0f);
|
|
std::fill(merge.begin() + 25, merge.begin() + 35, 1.0f);
|
|
check(merge, split, {{160,6400}});
|
|
std::cout << "FireRed VAD rules match official postprocessor fixtures\n";
|
|
return 0;
|
|
} catch (const std::exception & error) { std::cerr << error.what() << '\n'; return 1; }
|
|
}
|