def setUp(self):
self.bin = pj(program_path, "OnsetDetector")
self.activations = Activations(
pj(ACTIVATIONS_PATH, "sample.onsets_brnn.npz"))
self.result = np.loadtxt(
pj(DETECTIONS_PATH, "sample.onset_detector.txt"))
def setUp(self):
self.bin = pj(program_path, "MMBeatTracker")
self.activations = Activations(
pj(ACTIVATIONS_PATH, "sample.beats_blstm_mm.npz"))
self.result = np.loadtxt(
pj(DETECTIONS_PATH, "sample.mm_beat_tracker.txt"))
def setUp(self):
self.bin = pj(program_path, "LogFiltSpecFlux")
self.activations = Activations(
pj(ACTIVATIONS_PATH, "sample.log_filt_spec_flux.npz"))
self.result = np.loadtxt(
pj(DETECTIONS_PATH, "sample.log_filt_spec_flux.txt"))
def setUp(self):
self.bin = pj(program_path, "CRFBeatDetector")
self.activations = Activations(
pj(ACTIVATIONS_PATH, "sample.beats_blstm.npz"))
self.result = np.loadtxt(
pj(DETECTIONS_PATH, "sample.crf_beat_detector.txt"))
def setUp(self):
self.bin = pj(program_path, "ComplexFlux")
self.activations = Activations(
pj(ACTIVATIONS_PATH, "sample.complex_flux.npz"))
self.result = np.loadtxt(
pj(DETECTIONS_PATH, "sample.complex_flux.txt"))
"""
from __future__ import absolute_import, division, print_function
import unittest
from os.path import join as pj
from . import AUDIO_PATH, ACTIVATIONS_PATH
from madmom.audio.signal import FramedSignal
from madmom.features import Activations
from madmom.features.beats import *
from madmom.features.beats_hmm import *
from madmom.ml.hmm import HiddenMarkovModel
sample_file = pj(AUDIO_PATH, "sample.wav")
sample_lstm_act = Activations(pj(ACTIVATIONS_PATH, "sample.beats_lstm.npz"))
sample_blstm_act = Activations(pj(ACTIVATIONS_PATH, "sample.beats_blstm.npz"))
sample_downbeat_act = Activations(
pj(ACTIVATIONS_PATH, "sample.downbeats_blstm.npz"))
sample_pattern_features = Activations(
pj(ACTIVATIONS_PATH, "sample.gmm_pattern_tracker.npz"))
class TestRNNBeatProcessorClass(unittest.TestCase):
def setUp(self):
self.processor = RNNBeatProcessor()
def test_process_blstm(self):
# load bi-directional RNN models
beat_act = self.processor(sample_file)
self.assertTrue(np.allclose(beat_act, sample_blstm_act, atol=1e-5))
