File size: 1,111 Bytes
2ec0615
 
 
 
 
 
 
 
 
 
 
 
 
0ffb5f8
2ec0615
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import matplotlib.pyplot as plt
import librosa.display
from pretty_midi import PrettyMIDI


# Note: these functions are meant to be played within an interactive Python shell
# Please refer to the synth.ipynb for an example of how to use them


def get_music(midi_file):
    """
    Load a midi file and return the PrettyMIDI object and the audio signal
    """
    print(f"Getting MIDI music from: {midi_file}")
    music = PrettyMIDI(midi_file=midi_file)
    waveform = music.fluidsynth()
    return music, waveform


def show_piano_roll(music_notes, fs=100):
    """
    Show the piano roll of a music piece, with all instruments squashed onto a single 128xN matrix
    :param music_notes: PrettyMIDI object
    :param fs: sampling frequency
    """
    # get the piano roll
    piano_roll = music_notes.get_piano_roll(fs)
    print("Piano roll shape: {}".format(piano_roll.shape))

    # plot the piano roll
    plt.figure(figsize=(12, 4))
    librosa.display.specshow(piano_roll, sr=100, x_axis="time", y_axis="cqt_note")
    plt.colorbar()
    plt.title("Piano roll")
    plt.tight_layout()
    plt.show()