Squashed 'tinyml-modelmaker/' changes from 5d7c775e..4c5a0163

4c5a0163 updated json
REVERT: 5d7c775e updated

git-subtree-dir: tinyml-modelmaker
git-subtree-split: 4c5a0163bf49590282489979b5c259159f45bc32

Author: Adithya-Thonse

data/descriptions/description_timeseries.json

@@ -6,7 +6,7 @@
 valid_model_list = ['MotorFault_model_1_t', 'MotorFault_model_2_t', 'MotorFault_model_3_t',
                     'ArcFault_model_200_t', 'ArcFault_model_300_t', 'ArcFault_model_700_t', 'ArcFault_model_1400_t',
                     'TimeSeries_Generic_1k_t', 'TimeSeries_Generic_4k_t', 'TimeSeries_Generic_6k_t',
-                    'TimeSeries_Generic_13k_t', 'Lenet5', 'MobileNetV1_58k_NPU', 'PIRDetection_model_1_t']
+                    'TimeSeries_Generic_13k_t', 'Lenet5', 'MobileNetV1_58k_NPU', 'MobileNetV2_58k_NPU'. 'PIRDetection_model_1_t']
 soft_tinie_targets = ['c28_soft_int_in_int_out']
 hard_tinie_targets = ['c28_hard_int_in_int_out']
 valid_tinie_targets = soft_tinie_targets + hard_tinie_targets

scripts/get_model_performance_data.py

@@ -60,11 +60,17 @@ def get_target_module_from_task_type(task_type):
     vision_task_types = [
         ai_modules.vision.constants.TASK_TYPE_IMAGE_CLASSIFICATION,
     ]
+      # Audio task types
+    audio_task_types = [
+        ai_modules.audio.constants.TASK_TYPE_AUDIO_CLASSIFICATION,
+    ]
 
     if task_type in timeseries_task_types:
         return 'timeseries'
     elif task_type in vision_task_types:
         return 'vision'
+    elif task_type in audio_task_types:
+        return 'audio'
     else:
         return None
 
@@ -81,5 +87,6 @@ def get_task_category_type_from_task_type(task_type):
         ai_modules.timeseries.constants.TASK_TYPE_GENERIC_TS_ANOMALYDETECTION: ai_modules.timeseries.constants.TASK_CATEGORY_TS_ANOMALYDETECTION,
         ai_modules.timeseries.constants.TASK_TYPE_GENERIC_TS_FORECASTING: ai_modules.timeseries.constants.TASK_CATEGORY_TS_FORECASTING,
         ai_modules.vision.constants.TASK_TYPE_IMAGE_CLASSIFICATION: ai_modules.vision.constants.TASK_CATEGORY_IMAGE_CLASSIFICATION,
+        ai_modules.audio.constants.TASK_TYPE_AUDIO_CLASSIFICATION: ai_modules.audio.constants.TASK_CATEGORY_AUDIO_CLASSIFICATION,
     }
     return task_type_to_task_category_type_map.get(task_type)

tinyml_modelmaker/__init__.py

@@ -31,6 +31,7 @@
 import sys
 from . import timeseries
 from . import vision
+from . import audio
 
 def get_target_module(backend_name):
     this_module = sys.modules[__name__]

tinyml_modelmaker/ai_modules/__init__.py

@@ -0,0 +1,31 @@
+#################################################################################
+# Copyright (c) 2023-2026, Texas Instruments
+# All Rights Reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# * Redistributions of source code must retain the above copyright notice, this
+#   list of conditions and the following disclaimer.
+#
+# * Redistributions in binary form must reproduce the above copyright notice,
+#   this list of conditions and the following disclaimer in the documentation
+#   and/or other materials provided with the distribution.
+#
+# * Neither the name of the copyright holder nor the names of its
+#   contributors may be used to endorse or promote products derived from
+#   this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+#################################################################################
+
+from . import constants, datasets, runner, training

tinyml_modelmaker/ai_modules/audio/__init__.py

@@ -0,0 +1,2 @@
+from ...common.compilation import tinyml_benchmark
+