Home-IoT/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2024-11-22 15:08:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

[micro_wake_word] Fix VAD detection and modify detection computation (#7164)

Browse source
This commit is contained in:
Kevin Ahrendt 2024-07-30 16:57:51 -04:00 committed by GitHub
parent d7231fadb1
commit dff6884bed
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
esphome/components/micro_wake_word/streaming_model.cpp
View file

@ -148,7 +148,7 @@ WakeWordModel::WakeWordModel(const uint8_t *model_start, float probability_cutof
}; };
bool WakeWordModel::determine_detected() { bool WakeWordModel::determine_detected() {
int32_t sum = 0; uint32_t sum = 0;
for (auto &prob : this->recent_streaming_probabilities_) { for (auto &prob : this->recent_streaming_probabilities_) {
sum += prob; sum += prob;
} }
@ -175,12 +175,14 @@ VADModel::VADModel(const uint8_t *model_start, float probability_cutoff, size_t
}; };
bool VADModel::determine_detected() { bool VADModel::determine_detected() {
uint8_t max = 0; uint32_t sum = 0;
for (auto &prob : this->recent_streaming_probabilities_) { for (auto &prob : this->recent_streaming_probabilities_) {
max = std::max(prob, max); sum += prob;
} }
return max > this->probability_cutoff_; float sliding_window_average = static_cast<float>(sum) / static_cast<float>(255 * this->sliding_window_size_);
return sliding_window_average > this->probability_cutoff_;
} }
} // namespace micro_wake_word } // namespace micro_wake_word