Q2.py

#!/usr/bin/python
# -*- coding:utf-8 -*-
from glob import glob
import librosa

import utils

# Compute local onset autocorrelation
DB = 'Ballroom'
GENRE = [g.split('/')[2] for g in glob(DB + '/wav/*')]

# %% Q2
genres_p, genres_ALOTC = list(), list()

for g in GENRE:
    print('GENRE:', g)
    FILES = glob(DB + '/wav/' + g + '/*.wav')
    label, pred_t1, pred_t2, p_score, ALOTC_score = list(), list(), list(), list(), list()

    for f in FILES:
        f = f.replace('\\', '/')
        print('FILE:', f)

        # Read the labeled tempo
        bpm = float(utils.read_tempofile(DB, f))
        print('ground-truth tempo: ', bpm)
        label.append(bpm)

        # Estimate a static tempo
        sr, y = utils.read_wav(f)

        hop_length = 512
        oenv = librosa.onset.onset_strength(y=y, sr=sr, hop_length=hop_length)

        # tempogram = librosa.feature.tempogram(onset_envelope=oenv, sr=sr, hop_length=hop_length)

        # predict the tempo1(slower one), tempo2(faster one)
        # tempo1, tempo2 = librosa.beat.tempo(onset_envelope=oenv, sr=sr, hop_length=hop_length)
        tempo1, tempo2 = utils.tempo(onset_envelope=oenv, sr=sr, hop_length=hop_length)
        tempo1 = tempo1 * 3
        tempo2 = tempo2 * 3
        pred_t1.append(tempo1)
        pred_t2.append(tempo2)
        print(tempo1, tempo2)

        # p score
        s1 = tempo1/(tempo1+tempo2)
        s2 = 1.0 - s1
        print(s1, s2)
        p = s1 * utils.P_score(tempo1, bpm) + s2 * utils.P_score(tempo2, bpm)
        p_score.append(p)

        # ALOTC score
        ALOTC = utils.ALOTC(tempo1, tempo2, bpm)
        ALOTC_score.append(ALOTC)

        print(p, ALOTC)

    p_avg = sum(p_score)/len(p_score)
    ALOTC_avg = sum(ALOTC_score)/len(ALOTC_score)
    genres_p.append(p_avg)
    genres_ALOTC.append(ALOTC_avg)

    print('----------')

print(genres_p)
print(genres_ALOTC)
print()

print("***** Q2 *****")
print("Genre          \tP-score    \tALOTC score")
for g in range(len(GENRE)):
    print("{:13s}\t{:8.2f}\t{:8.2f}".format(GENRE[g], genres_p[g], genres_ALOTC[g]))
print('----------')
print("Overall P-score:\t{:.2f}".format(sum(genres_p)/len(genres_p)))
print("Overall ALOTC score:\t{:.2f}".format(sum(genres_ALOTC)/len(genres_ALOTC)))