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