Rename all wake words

This commit is contained in:
xysun 2024-01-16 16:13:24 +08:00
parent 11cfe7a1b1
commit ce2a34ef79
28 changed files with 37 additions and 40 deletions

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

View File

@ -1 +1 @@
wakeNet7_v1_xiaoaitongxue_6_0.97_0.92
wakeNet7_v1_小爱同学_6_0.97_0.92

View File

@ -1 +1 @@
wakeNet7Q8_v1h8_xiaoaitongxue_5_0.97_0.98
wakeNet7Q8_v1h8_小爱同学_5_0.97_0.98

View File

@ -1 +1 @@
wakeNet8_v5h8_alexa_5_0.57_0.60
wakeNet8_v5h8_Alexa_5_0.57_0.60

View File

@ -1 +1 @@
WakeNet8_v3h8_hiesp_5_0.60_0.616
WakeNet8_v3h8_Hi,ESP_5_0.60_0.616

View File

@ -1 +1 @@
WakeNet8_v3h8_hilexin_5_0.625_0.635
WakeNet8_v3h8_嗨乐鑫_5_0.625_0.635

View File

@ -1,2 +1 @@
# (neural network type)_(model data version)_(lable1_detection windown length_threshold for 90%_threshold for 95%)_(lable2 ...)_...
wakenet9l_v4h8_alexa_3_0.640_0.650
wakenet9l_v4h8_Alexa_3_0.640_0.650

View File

@ -1,2 +1 @@
# (neural network type)_(model data version)_(lable1_detection windown length_threshold for 90%_threshold for 95%)_(lable2 ...)_...
wakenet9l_tts1h8_computer_3_0.648_0.650
wakenet9l_tts1h8_Computer_3_0.648_0.650

View File

@ -1 +1 @@
WakeNet9_v1h24_xiaoaitongxue_3_0.620_0.627
WakeNet9_v1h24_小爱同学_3_0.620_0.627

View File

@ -1,2 +1 @@
# (neural network type)_(model data version)_(lable1_detection windown length_threshold for 90%_threshold for 95%)_(lable2 ...)_...
wakenet9l_tts1h8_heywillow_3_0.644_0.650
wakenet9l_tts1h8_Hey,Willow_3_0.644_0.650

View File

@ -1 +1 @@
wakeNet9_v1h24_hiesp_3_0.63_0.635
wakeNet9_v1h24_Hi,ESP_3_0.63_0.635

View File

@ -1 +1 @@
wakenet9_v1h24_hilexin_3_0.608_0.615
wakenet9_v1h24_嗨,乐鑫_3_0.608_0.615

View File

@ -1,2 +1 @@
# (neural network type)_(model data version)_(lable1_detection windown length_threshold for 90%_threshold for 95%)_(lable2 ...)_...
wakenet9l_v2h8_himfive_3_0.640_0.645
wakenet9l_v2h8_Hi,M Five_3_0.640_0.645

View File

@ -1,2 +1,2 @@
# (neural network type)_(model data version)_(lable1_detection windown length_threshold for 90%_threshold for 95%)_(lable2 ...)_...
wakenet9l_tts1h8_jarvis_3_0.627_0.632
wakenet9l_tts1h8_Jarvis_3_0.627_0.632

View File

@ -1 +1 @@
wakenet9_v1h12_nihaoxiaozhi_3_0.585_0.595
wakenet9_v1h12_你好小智_3_0.585_0.595

View File

@ -1 +1 @@
WakeNet9_v1h24_xiaoaitongxue_3_0.620_0.627
WakeNet9_v1h24_小爱同学_3_0.620_0.627

View File

@ -28,33 +28,29 @@ void set_model_base_path(const char *base_path)
char *get_model_info(char *data, int size)
{
char *temp = malloc((size+1)*sizeof(char));
char *buffer = temp;
char *model_info = NULL;
memcpy(buffer, data, size);
buffer[size] = '\0';
printf("buffer: %s\n", buffer);
//Prase
//if the line starts with '#', the line is a comment
//else the line is model information
while(1) {
if (*buffer == '#') {
while (*buffer != '\n') {
buffer ++;
while(size > 0) {
if (*data == '#') {
while (*data != '\n' && size > 1) {
data ++;
size --;
}
buffer ++;
data ++;
size --;
continue;
} else if (buffer != NULL && buffer[0]) {
int info_len = strlen(buffer);
model_info = (char*) malloc((info_len+1)*sizeof(char));
memcpy(model_info, buffer, info_len);
model_info[info_len] = '\0';
printf("model info: %s\n", model_info);
} else if (data != NULL && size > 0) {
model_info = (char*)malloc((size + 1) * sizeof(char));
memcpy(model_info, data, size);
if (model_info[size - 1] == '\n') {
model_info[size - 1] = '\0';
}
model_info[size] = '\0';
break;
}
}
free(temp);
return model_info;
}
@ -85,7 +81,6 @@ char *get_wake_words_from_info(char *model_info)
wake_words[word_len-1] = '\0';
}
}
printf("token %d: %s\n", i, token);
token = strtok(NULL, "_");
i++;
}
@ -367,6 +362,7 @@ srmodel_list_t *srmodel_mmap_init(const esp_partition_t *partition)
for (int i = 0; i < models->num; i++) {
srmodel_data_t *model_data = (srmodel_data_t *) malloc(sizeof(srmodel_data_t));
models->model_info[i] = NULL;
// read model name
models->model_name[i] = (char *)malloc((strlen(data) + 1) * sizeof(char));
strcpy(models->model_name[i], data);
@ -391,7 +387,6 @@ srmodel_list_t *srmodel_mmap_init(const esp_partition_t *partition)
int size = read_int32(data);
data += int_len;
model_data->sizes[j] = size;
printf("file %s\n", model_data->files[j]);
// read model info
if (strcmp(model_data->files[j], "_MODEL_INFO_") == 0) {
@ -422,11 +417,14 @@ void srmodel_mmap_deinit(srmodel_list_t *models)
free(models->model_data[i]->sizes);
free(models->model_data[i]);
free(models->model_name[i]);
if (models->model_info[i] != NULL)
free(models->model_info[i]);
}
}
free(models->model_data);
free(models->model_name);
free(models->mmap_handle);
free(models->model_info);
free(models);
}
models = NULL;
@ -558,6 +556,7 @@ void srmodel_sdcard_deinit(srmodel_list_t *models)
if (models->num > 0) {
for (int i = 0; i < models->num; i++) {
free(models->model_name[i]);
free(models->model_info[i]);
}
}
free(models);

View File

@ -54,6 +54,7 @@ TEST_CASE("wakenet create/destroy API & memory leak", "[wn]")
models = esp_srmodel_init("model");
model_name = esp_srmodel_filter(models, ESP_WN_PREFIX, NULL);
wakenet = (esp_wn_iface_t*)esp_wn_handle_from_name(model_name);
// char *wake_words = esp_srmodel_get_wake_words(models, model_name);
printf("create ...\n");
// typedef enum {
@ -68,6 +69,7 @@ TEST_CASE("wakenet create/destroy API & memory leak", "[wn]")
printf("destroy ...\n");
wakenet->destroy(model_data);
// free(wake_words);
esp_srmodel_deinit(models);
last_end_size = heap_caps_get_free_size(MALLOC_CAP_8BIT);

View File

@ -1,4 +1,4 @@
# Espressif ESP32 Partition Table
# Name, Type, SubType, Offset, Size
factory, app, factory, 0x010000, 8000k
model, data, , , 7000K,
model, data, , , 6500K,

1 # Espressif ESP32 Partition Table
2 # Name, Type, SubType, Offset, Size
3 factory, app, factory, 0x010000, 8000k
4 model, data, , , 7000K, model, data, , , 6500K,