#include <streaming_model.h>
Inheritance diagram for esphome::micro_wake_word::VADModel:
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} |
Detailed Description
Definition at line 69 of file streaming_model.h.
Constructor & Destructor Documentation
◆ VADModel()
| 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.
Member Function Documentation
◆ determine_detected()
|
overridevirtual |
Checks for voice activity by comparing the max probability in the sliding window with the probability cutoff.
- Returns
- True if voice activity is detected, false otherwise
Implements esphome::micro_wake_word::StreamingModel.
Definition at line 177 of file streaming_model.cpp.
◆ log_model_config()
|
overridevirtual |
Implements esphome::micro_wake_word::StreamingModel.
Definition at line 20 of file streaming_model.cpp.
The documentation for this class was generated from the following files:
- /opt/build/esphome/esphome/components/micro_wake_word/streaming_model.h
- /opt/build/esphome/esphome/components/micro_wake_word/streaming_model.cpp
