Summary

openav/api/preprocess_audio.py

@@ -37,7 +37,7 @@
 from openav import rsrs  # Ресурсы библиотеки
 
 from openav.modules.core.logging import ARG_PATH_TO_LOGS
-from openav.modules.lab.audio import SAMPLING_RATE_MS, PAD_MODE_MS, DPI, COLOR_GRADIENTS
+from openav.modules.lab.audio import SAMPLING_RATE_MS, PAD_MODE_MS, DPI, COLOR_GRADIENTS, EXT_AUDIO
 
 
 # ######################################################################################################################
@@ -74,7 +74,7 @@ class RunPreprocessAudio(MessagesPreprocessAudio):
     def __post_init__(self):
         super().__post_init__()  # Выполнение конструктора из суперкласса
 
-        self._all_layer_in_yaml = 17  # Общее количество настроек в конфигурационном файле
+        self._all_layer_in_yaml = 18  # Общее количество настроек в конфигурационном файле
 
         #  Регистратор логирования с указанным именем
         self._logger_run_preprocess_audio: logging.Logger = logging.getLogger(__class__.__name__)
@@ -166,8 +166,15 @@ def _valid_yaml_config(self, config: Dict[str, Union[str, bool, int, float]], ou
             # 1. Скрытие метаданных
             # 2. Скрытие версий установленных библиотек
             # 3. Включение установки отступов с обеих сторон относительно центра аудиодорожки
-            # 4. Сохранение сырых данных мел-спектрограммы в формате .npy
-            if key == "hide_metadata" or key == "hide_libs_vers" or key == "center" or key == "save_raw_data":
+            # 4. Очистка директории для сохранения аудиоданных после предобработки
+            # 5. Сохранение сырых данных мел-спектрограммы в формате .npy
+            if (
+                key == "hide_metadata"
+                or key == "hide_libs_vers"
+                or key == "center"
+                or key == "clear_dir_audio"
+                or key == "save_raw_data"
+            ):
                 # Проверка значения
                 if type(val) is not bool:
                     continue
@@ -193,12 +200,13 @@ def _valid_yaml_config(self, config: Dict[str, Union[str, bool, int, float]], ou
                 # Проход по всем подразделам текущего раздела
                 for v in val:
                     # Проверка значения
-                    if type(v) is not str or not v:
+                    if type(v) is not str or not v or (v in EXT_AUDIO) is False:
+                        curr_valid_layer_2 += 100
                         continue
 
                     curr_valid_layer_2 += 1
 
-                if curr_valid_layer_2 > 0:
+                if curr_valid_layer_2 <= len(EXT_AUDIO):
                     curr_valid_layer += 1
 
             # 1. Путь к директории набора данных
@@ -350,7 +358,7 @@ def run(self, metadata: ModuleType = openav, resources: ModuleType = rsrs, out:
             out (bool): Печатать процесс выполнения
 
         Returns:
-             bool: **True** если детектирование речевой активности в аудиовизуальном сигнале произведено успешно,
+             bool: **True** если предобработка речевых аудиоданных произведена успешно,
                    в обратном случае **False**
         """