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