ESPHome
2024.10.2
|
#include <streaming_model.h>
Public Member Functions | |
VADModel (const uint8_t *model_start, float probability_cutoff, size_t sliding_window_size, size_t tensor_arena_size) | |
void | log_model_config () override |
bool | determine_detected () override |
Checks for voice activity by comparing the max probability in the sliding window with the probability cutoff. More... | |
Public Member Functions inherited from esphome::micro_wake_word::StreamingModel | |
bool | perform_streaming_inference (const int8_t features[PREPROCESSOR_FEATURE_SIZE]) |
void | reset_probabilities () |
Sets all recent_streaming_probabilities to 0. More... | |
bool | load_model (tflite::MicroMutableOpResolver< 20 > &op_resolver) |
Allocates tensor and variable arenas and sets up the model interpreter. More... | |
void | unload_model () |
Destroys the TFLite interpreter and frees the tensor and variable arenas' memory. More... | |
Additional Inherited Members | |
Protected Attributes inherited from esphome::micro_wake_word::StreamingModel | |
uint8_t | current_stride_step_ {0} |
float | probability_cutoff_ |
size_t | sliding_window_size_ |
size_t | last_n_index_ {0} |
size_t | tensor_arena_size_ |
std::vector< uint8_t > | recent_streaming_probabilities_ |
const uint8_t * | model_start_ |
uint8_t * | tensor_arena_ {nullptr} |
uint8_t * | var_arena_ {nullptr} |
std::unique_ptr< tflite::MicroInterpreter > | interpreter_ |
tflite::MicroResourceVariables * | mrv_ {nullptr} |
tflite::MicroAllocator * | ma_ {nullptr} |
Definition at line 69 of file streaming_model.h.
esphome::micro_wake_word::VADModel::VADModel | ( | const uint8_t * | model_start, |
float | probability_cutoff, | ||
size_t | sliding_window_size, | ||
size_t | tensor_arena_size | ||
) |
Definition at line 168 of file streaming_model.cpp.
|
overridevirtual |
Checks for voice activity by comparing the max probability in the sliding window with the probability cutoff.
Implements esphome::micro_wake_word::StreamingModel.
Definition at line 177 of file streaming_model.cpp.
|
overridevirtual |
Implements esphome::micro_wake_word::StreamingModel.
Definition at line 20 of file streaming_model.cpp.