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