def extract_melgrams(list_path, MULTIFRAMES, process_all_song,
num_songs_genre):
melgrams = np.zeros((0, 1, 96, 1366), dtype=np.float32)
song_paths = open(list_path, 'r').read().splitlines()
labels = list()
num_frames_total = list()
for song_ind, song_path in enumerate(song_paths):
print(song_path)
if MULTIFRAMES:
melgram = ap.compute_melgram_multiframe(song_path,
process_all_song)
num_frames = melgram.shape[0]
num_frames_total.append(num_frames)
print('num frames:', num_frames)
if num_songs_genre != '':
index = int(floor(song_ind / num_songs_genre))
for i in range(0, num_frames):
labels.append(index)
else:
pass
else:
melgram = ap.compute_melgram(song_path)
melgrams = np.concatenate((melgrams, melgram), axis=0)
if num_songs_genre != '':
return melgrams, labels, num_frames_total
else:
return melgrams, num_frames_total
def extract_melgrams(song_paths, MULTIFRAMES, process_all_song,
num_songs_genre):
melgrams = np.zeros((0, 1, 96, 1366), dtype=np.float32)
#song_paths = open(list_path, 'r').read().splitlines()
labels = list()
num_frames_total = list()
for song_ind, song_path in enumerate(song_paths):
print song_path
if MULTIFRAMES:
melgram = ap.compute_melgram_multiframe(song_path,
process_all_song)
num_frames = melgram.shape[0]
num_frames_total.append(num_frames)
print 'num frames:', num_frames
else:
melgram = ap.compute_melgram(song_path)
melgrams = np.concatenate((melgrams, melgram), axis=0)
return melgrams, num_frames_total