-
Notifications
You must be signed in to change notification settings - Fork 0
/
Q4-2.py
95 lines (87 loc) · 2.67 KB
/
Q4-2.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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/usr/bin/python
# -*- coding:utf-8 -*-
from glob import glob
from collections import defaultdict
import librosa
import numpy as np
from scipy.stats import pearsonr
# %%
import utils # self-defined utils.py file
DB = 'GiantSteps'
if DB == 'GTZAN': # dataset with genre label classify at parent directory
FILES = glob(DB + '/wav/*/*.wav')
# print(FILES)
else:
FILES = glob(DB + '/wav/*.wav')
# print(FILES)
GENRE = [g.split('/')[2] for g in glob(DB + '/wav/*')]
print(GENRE)
n_fft = 100 # (ms)
hop_length = 25 # (ms)
# %% Q4-2
if DB == 'GTZAN':
label, pred = defaultdict(list), defaultdict(list)
else:
label, pred = list(), list()
chromagram = list()
gens = list()
for f in FILES:
f = f.replace('\\', '/')
print("file: ", f)
content = utils.read_keyfile(f, '*.txt')
content = utils.generalize_key(content)
content = utils.str_to_lerch(content)
# if (int(content) < 0): continue # skip saving if key not found
if DB == 'GTZAN':
gen = f.split('/')[2]
label[gen].append(utils.LABEL[int(content)])
gens.append(gen)
else:
label.append(utils.LABEL[content])
sr, y = utils.read_wav(f)
cxx = librosa.feature.chroma_stft(y=y, sr=sr)
chromagram.append(cxx) # store into list for further use
chroma_vector = np.sum(cxx, axis=1)
key_ind = np.where(chroma_vector == np.amax(chroma_vector))
key_ind = int(key_ind[0])
# print('key index: ', key_ind)
chroma_vector = utils.rotate(chroma_vector.tolist(), 12 - key_ind)
# print('chroma_vector: ', chroma_vector)
KS = {"major": [6.35, 2.23, 3.48, 2.33, 4.38, 4.09, 2.52, 5.19, 2.39, 3.66, 2.29, 2.88],
"minor": [6.33, 2.68, 3.52, 5.38, 2.60, 3.53, 2.54, 4.75, 3.98, 2.69, 3.34, 3.17]}
r_co_major = pearsonr(chroma_vector, KS["major"])
r_co_minor = pearsonr(chroma_vector, KS["minor"])
# print(r_co_major[0])
# print(r_co_minor[0])
mode = ''
if (r_co_major[0] > r_co_minor[0]):
mode = key_ind
else:
mode = key_ind + 12
mode = utils.lerch_to_str(mode)
# print('mode', mode)
if DB == 'GTZAN':
pred[gen].append(mode)
else:
pred.append(mode) # you may ignore this when starting with GTZAN dataset
# print(pred[gen])
print("***** Q4-2 *****")
# TODO: Calculate the accuracy for all file.
# Hint1: Use label_list and pred_list.
correct_all = 0
for acc_len in range(len(label)):
if label[acc_len] == pred[acc_len]:
correct_all += 1
try:
acc_all = correct_all / len(label)
except ZeroDivisionError:
acc_all = 0
##########
print("----------")
print("Overall accuracy:\t{:.2f}%".format(acc_all))
'''
GiantSteps
***** Q4-2 *****
----------
Overall accuracy: 0.28%
'''