Summary

DmitryRyumin · Jan 27, 2025 · 6fee33a · 6fee33a
1 parent 84c7eb3
commit 6fee33a
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 9 deletions.
diff --git a/openav/__init__.py b/openav/__init__.py
@@ -18,7 +18,7 @@
 __uri__ = "https://github.com/DmitryRyumin/openav"
 
 __version__ = "1.0"
-__release__ = __version__ + ".0-a17"
+__release__ = __version__ + ".0-a18"
 
 __author__ru__ = "Рюмин Дмитрий, Иванько Денис, Шилов Николай, Маркитантов Максим, Карпов Алексей"
 __author__en__ = "Dmitry Ryumin, Denis Ivanko, Nikolay Shilov, Maxim Markitantov, Alexey Karpov"
@@ -31,4 +31,4 @@
 __maintainer_email__ = "dl_03.03.1991@mail.ru"
 
 __license__ = "MIT"
-__copyright__ = "2022-24. Dmitry Ryumin, Denis Ivanko, Nikolay Shilov, Maxim Markitantov, Alexey Karpov"
+__copyright__ = "2022-25. Dmitry Ryumin, Denis Ivanko, Nikolay Shilov, Maxim Markitantov, Alexey Karpov"
diff --git a/openav/modules/lab/audiovisual.py b/openav/modules/lab/audiovisual.py
@@ -59,6 +59,38 @@
 EXTH_MODELS: str = "pth"
 OPTIMIZERS: List[str] = ["adam", "adamw", "sgd", "lion"]
 REQUIRED_GRAD: List[str] = ["none", "a", "v", "av"]
+CLASSES_TEST: List[str] = [
+    "1_Позвонить",
+    "2_Набрать_номер",
+    "3_Отправить_сообщение",
+    "6_Завершить_вызов",
+    "7_Радио",
+    "8_Музыка",
+    "9_Воспроизвести",
+    "11_Случа--и--ны--и--_выбор",
+    "12_Отключить_случа--и--ны--и--_выбор",
+    "13_Повтор",
+    "16_Избегать_платных_дорог",
+    "17_Карта",
+    "19_Предыдущие_места_назначения",
+    "20_Как_там_на_дорогах",
+    "22_Надолго_пробка",
+    "24_На_работу",
+    "25_Максимальное_увеличение",
+    "27_Остановить_маршрут",
+    "28_Возобновить_маршрут",
+    "29_Сколько_мне_еще_ехать",
+    "30_Во_сколько_я_приеду",
+    "32_Сбросить_маршрут",
+    "34_На--и--ти_больницу",
+    "36_На--и--ти_аптеку",
+    "37_На--и--ти_банк",
+    "39_На--и--ти_ресторан",
+    "42_На--и--ти_железнодорожны--и--_вокзал",
+    "47_Нет",
+    "48_Предыдущая",
+    "49_Следующая",
+]
 
 
 # ######################################################################################################################
@@ -372,22 +404,25 @@ def train_audiovisual(
 
             # Проход по всем вложенным директориям
             for nested_path in hierarchy_from_paths:
-                if nested_path[0].lower() in classes:
+                nested_path_0 = nested_path[0].replace("--и--", "й").lower()
+                if nested_path_0 in classes:
                     # Формирование списка с видеофайлами
                     for p in Path(os.path.join(self.path_to_dataset, *reversed(nested_path))).glob("*"):
                         # Добавление текущего пути к видеофайлу в список
                         if p.suffix.lower().replace(".", "") in EXT_AV_VIDEO:
-                            [index for index, cls in enumerate(classes) if cls.lower() == nested_path[0].lower()]
+                            [index for index, cls in enumerate(classes) if cls.lower() == nested_path_0]
 
                             if nested_path[-1] == subfolders[SUBFOLDERS[0]]:
                                 path_train.append(p.resolve())
-                                lb_train.append(classes.index(nested_path[0].lower()))
+                                lb_train.append(classes.index(nested_path_0))
                             elif nested_path[-1] == subfolders[SUBFOLDERS[1]]:
-                                path_val.append(p.resolve())
-                                lb_val.append(classes.index(nested_path[0].lower()))
+                                if (nested_path[0] in CLASSES_TEST) is True:
+                                    path_val.append(p.resolve())
+                                    lb_val.append(classes.index(nested_path_0))
                             elif nested_path[-1] == subfolders[SUBFOLDERS[2]]:
-                                path_test.append(p.resolve())
-                                lb_test.append(classes.index(nested_path[0].lower()))
+                                if (nested_path[0] in CLASSES_TEST) is True:
+                                    path_test.append(p.resolve())
+                                    lb_test.append(classes.index(nested_path_0))
                             else:
                                 pass