From 7982e1a8b901a28da7726291dbe8a50447c33039 Mon Sep 17 00:00:00 2001
From: Sting <loic.allegre@ensiie.fr>
Date: Thu, 22 Jun 2023 14:20:04 +0200
Subject: [PATCH] Training data preprocess 2

---
 music_processor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/music_processor.py b/music_processor.py
index 44bd4cd..580443b 100644
--- a/music_processor.py
+++ b/music_processor.py
@@ -267,7 +267,7 @@ def music_for_train(serv, verbose=False, nhop=512, nffts=[1024, 2048, 4096], mel
         if verbose:
             print(songplace)
         
-        song = Audio(glob(songplace+"/*.wav")[0])
+        song = Audio(glob(songplace+"/*.ogg")[0])
         song.import_ass(glob(songplace+"/*.ass")[-1])
         song.data = (song.data[:, 0]+song.data[:, 1])/2
         songs.append(song)
@@ -280,7 +280,7 @@ def music_for_train(serv, verbose=False, nhop=512, nffts=[1024, 2048, 4096], mel
 
 def music_for_test(serv, deletemusic=True, verbose=False):
 
-    song = Audio(glob(serv+"/*.wav")[0], stereo=False)
+    song = Audio(glob(serv+"/*.ogg")[0], stereo=False)
     # song.import_tja(glob(serv+"/*.tja")[-1])
     song.feats = fft_and_melscale(song, include_zero_cross=False)
     with open('./data/pickles/test_data.pickle', mode='wb') as f:
-- 
GitLab