feat: Add Jarvis and Computer wakenet trained by TTS samples

CHANGELOG.md

@@ -7,6 +7,7 @@ If you can not map model partition successfully, please check the left free stor
 ## unreleased
 - Add Chinese MultiNet7 models
 - Add first Noise Suppression model: nsnet1
+- Add wakenet model trained by TTS samples: Jarvis and Computer
 
 ## 1.5.1
 - Reduce Internal RAM of multinet7

Kconfig.projbuild

@@ -107,6 +107,14 @@ choice SR_WN_MODEL_LOAD
     config SR_WN_WN9_NIHAOXIAOZHI
         bool "nihaoxiaozhi (wn9_nihaoxiaozhi)"
         depends on IDF_TARGET_ESP32S3
+    
+    config SR_WN_WN9_JARVIS_TTS
+        bool "jarvis (wn9_jarvis_tts)"
+        depends on IDF_TARGET_ESP32S3
+    
+    config SR_WN_WN9_COMPUTER_TTS
+        bool "computer (wn9_computer_tts)"
+        depends on IDF_TARGET_ESP32S3
 
     config SR_WN_WN9_CUSTOMWORD
         bool "customized word (wn9_customword)"

model/movemodel.py

@@ -51,26 +51,10 @@ def copy_wakenet_from_sdkconfig(model_path, sdkconfig_path, target_path):
         models.append('wn9_himfive')
     if "CONFIG_SR_WN_WN9_NIHAOXIAOZHI" in models_string:
         models.append('wn9_nihaoxiaozhi')
-    if "CONFIG_SR_WN_WN9_CUSTOMWORD" in models_string:
-        models.append('wn9_customword')
-
-    for item in models:
-        shutil.copytree(model_path + '/wakenet_model/' + item, target_path+'/'+item)
-
-def copy_multinet_from_sdkconfig(model_path, sdkconfig_path, target_path):
-    """
-    Copy multinet model from model_path to target_path based on sdkconfig
-    """
-    with io.open(sdkconfig_path, "r") as f:
-        models_string = ''
-        for label in f:
-            label = label.strip("\n")
-            if 'CONFIG_SR_MN' in label and label[0] != '#':
-                models_string += label
-
-    models = []
-    if "CONFIG_SR_WN_WN9_NIHAOXIAOZHI" in models_string:
-        models.append('wn9_nihaoxiaozhi')
+    if "CONFIG_SR_WN_WN9_JARVIS_TTS" in models_string:
+        models.append('wn9_jarvis_tts')
+    if "CONFIG_SR_WN_WN9_COMPUTER_TTS" in models_string:
+        models.append('wn9_computer_tts')
     if "CONFIG_SR_WN_WN9_CUSTOMWORD" in models_string:
         models.append('wn9_customword')
 

model/wakenet_model/wn9_computer_tts/_MODEL_INFO_

@@ -0,0 +1,2 @@
+# (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

model/wakenet_model/wn9_jarvis_tts/_MODEL_INFO_

@@ -0,0 +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