Merge branch 'it18218640' into 'master'

It18218640

See merge request !99

backend/IT18218640/convert.py

+import pydub
+from pydub import AudioSegment
+import subprocess
+def main():
+
+
+    # files
+    src = "src/Blue1643090283.mp3"
+    dst = "dst/test.wav"
+
+    # subprocess.call(['ffmpeg', '-i', 'src/Blue1643090283.mp3',
+    #                  'dst/test.wav'])
+
+    # convert wav to mp3
+    audSeg = AudioSegment.from_mp3(src).export(dst, format="wav")
+
+    # print(f"{audSeg}")
+
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file

backend/IT18218640/train.py

@@ -12,7 +12,7 @@ LEARNING_RATE = 0.0001
 EPOCHS = 60
 BATCH_SIZE = 32
 
-NUM_KEYWORDS = 10
+NUM_KEYWORDS = 6
 
 def load_dataset(data_path):
 
@@ -67,6 +67,12 @@ def build_model(input_shape, learning_rate, error="sparse_categorical_crossentro
     model.add(keras.layers.Dense(64, activation="relu"))
     model.add(keras.layers.Dropout(0.3))
 
+    # model.add(keras.layers.Flatten())
+    # model.add(keras.layers.Dense(128, activation="relu"))
+    # model.add(keras.layers.Dense(64, activation="relu"))
+    # model.add(keras.layers.Dropout(0.3))
+    # model.add(keras.layers.Dense(16, activation="relu"))
+
     # softmax classifier
     model.add(keras.layers.Dense(NUM_KEYWORDS, activation="softmax"))