update

examples/vm_sound_classification/step_1_prepare_data.py

@@ -37,6 +37,18 @@ def get_dataset(args):
     file_dir = Path(args.file_dir)
     file_dir.mkdir(exist_ok=True)
 
+    label2_map = {
+        "bell": "non_voice",
+        "white_noise": "non_voice",
+        "low_white_noise": "non_voice",
+        "high_white_noise": "non_voice",
+        "music": "non_voice",
+        "mute": "non_voice",
+        "noise": "non_voice",
+        "noise_mute": "non_voice",
+        "voice": "voice",
+        "voicemail": "voice",
+    }
     # label3_map = {
     #     "bell": "voicemail",
     #     "white_noise": "mute",
@@ -49,18 +61,18 @@ def get_dataset(args):
     #     "voice": "voice_or_noise",
     #     "voicemail": "voicemail",
     # }
-    label8_map = {
-        "bell": "bell",
-        "white_noise": "white_noise",
-        "low_white_noise": "white_noise",
-        "high_white_noise": "white_noise",
-        "music": "music",
-        "mute": "mute",
-        "noise": "noise",
-        "noise_mute": "noise_mute",
-        "voice": "voice",
-        "voicemail": "voicemail",
-    }
+    # label8_map = {
+    #     "bell": "bell",
+    #     "white_noise": "white_noise",
+    #     "low_white_noise": "white_noise",
+    #     "high_white_noise": "white_noise",
+    #     "music": "music",
+    #     "mute": "mute",
+    #     "noise": "noise",
+    #     "noise_mute": "noise_mute",
+    #     "voice": "voice",
+    #     "voicemail": "voicemail",
+    # }
 
     result = list()
     for filename_pattern in filename_patterns:
@@ -74,10 +86,10 @@ def get_dataset(args):
             folder = filename.parts[-2]
             country = filename.parts[-4]
 
-            if folder not in label8_map.keys():
+            if folder not in label2_map.keys():
                 continue
 
-            labels = label8_map[folder]
+            labels = label2_map[folder]
 
             random1 = random.random()
             random2 = random.random()