update

examples/vm_sound_classification/step_1_prepare_data.py

@@ -37,18 +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",
-    }
+    # 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",
@@ -61,6 +61,18 @@ def get_dataset(args):
     #     "voice": "voice_or_noise",
     #     "voicemail": "voicemail",
     # }
+    label4_map = {
+        "bell": "voicemail",
+        "white_noise": "mute",
+        "low_white_noise": "mute",
+        "high_white_noise": "mute",
+        # "music": "music",
+        "mute": "mute",
+        "noise": "noise",
+        "noise_mute": "noise",
+        "voice": "voice_or_noise",
+        "voicemail": "voicemail",
+    }
     # label8_map = {
     #     "bell": "bell",
     #     "white_noise": "white_noise",
@@ -86,10 +98,10 @@ def get_dataset(args):
             folder = filename.parts[-2]
             country = filename.parts[-4]
 
-            if folder not in label2_map.keys():
+            if folder not in label4_map.keys():
                 continue
 
-            labels = label2_map[folder]
+            labels = label4_map[folder]
 
             random1 = random.random()
             random2 = random.random()