-
Notifications
You must be signed in to change notification settings - Fork 1
/
createLabels.py
33 lines (26 loc) · 877 Bytes
/
createLabels.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import pydub
import os
import numpy as np
from pydub import AudioSegment
from pydub.audio_segment import extract_wav_headers
import csv
files_path = 'genres/bigger_cut'
paths = np.array([])
labels = np.array([])
for root, dirs, files in os.walk(files_path):
for file in files:
s = ''
for i in file:
if (i == '.' or i == '_'):
break
s+=i
tempPath = root + '/' + file
paths = np.append(paths, tempPath)
labels = np.append(labels, s)
final = np.concatenate((np.reshape(paths,(len(paths),1) ), np.reshape(labels,(len(labels),1) )),axis = 1)
header = np.array(['path','genre'])
with open("updated_gtzan.csv", "w", newline='') as f:
writer = csv.writer(f, delimiter=',')
writer.writerow(header)
for l in final:
writer.writerow(l)