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)
# 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))