示例#1
0
if DB == 'GTZAN':
    label, pred = defaultdict(list), defaultdict(list)
else:
    label, pred = list(), list()
chromagram = list()
gens = list()

idx = 0
gamma = 1
while idx < 4 and gamma < 1001:
    print('gamma', gamma)
    for f in tqdm(FILES):
        f = f.replace('\\', '/')
        # print("file: ", f)
        content = utils.read_keyfile(f, '*.lerch.txt')
        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)

        # gamma = input("gamma (1, 10, 100, 1000): ")
        cxx = np.log(1 +
                     gamma * np.abs(librosa.feature.chroma_cens(y=y, sr=sr)))
        chromagram.append(cxx)  # store into list for further use
        chroma_vector = np.sum(cxx, axis=1)
示例#2
0
# GENRE.remove('classical')
# # print(GENRE)
n_fft = 100  # (ms)
hop_length = 25  # (ms)

if DB == 'GTZAN':
    label, pred = defaultdict(list), defaultdict(list)
else:
    label, pred = list(), list()
chromagram = list()
gens = list()

for f in tqdm(FILES):
    f = f.replace('\\', '/')
    # print("file: ", f)
    content = utils.read_keyfile(f, '*.key')
    if (len(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(content)

    sr, y = utils.read_wav(f)

    cxx = librosa.feature.chroma_cqt(y=y, sr=sr)
    chromagram.append(cxx)  # store into list for further use
    chroma_vector = np.sum(cxx, axis=1)
    # print(chroma_vector)
    key_ind = np.where(chroma_vector == np.amax(chroma_vector))