simplify a little midi.py module
This commit is contained in:
parent
0a297e331d
commit
4aa5916346
80
project/midi.py
Normal file
80
project/midi.py
Normal file
@ -0,0 +1,80 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import settings
|
||||||
|
import pypianoroll as roll
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
import numpy as np
|
||||||
|
import os
|
||||||
|
from tqdm import tqdm
|
||||||
|
from math import floor
|
||||||
|
import sys
|
||||||
|
|
||||||
|
def to_samples(midi_file_path, midi_res=settings.midi_resolution):
|
||||||
|
|
||||||
|
# this function export a samples from midi file:
|
||||||
|
# and for every track in midi file chopped pianoroll
|
||||||
|
# for a samples of given beat_lenth (midi_res)
|
||||||
|
# every track is single line
|
||||||
|
|
||||||
|
print('Exporting samples from: {}'.format(midi_file_path))
|
||||||
|
|
||||||
|
all_beats = np.empty((0, settings.midi_resolution, 128))
|
||||||
|
|
||||||
|
for track in roll.Multitrack(midi_file_path).tracks:
|
||||||
|
print('Track: {}'.format(track.name))
|
||||||
|
if not track.is_drum:
|
||||||
|
number_of_beats = floor(track.pianoroll.shape[0] / midi_res)
|
||||||
|
track_pianoroll = track.pianoroll[: number_of_beats * midi_res]
|
||||||
|
track_beats = track_pianoroll.reshape(number_of_beats, midi_res, 128)
|
||||||
|
all_beats = np.concatenate([track_beats, all_beats], axis=0)
|
||||||
|
|
||||||
|
print('Exported {} samples of {}'.format(number_of_beats, settings.midi_program[track.program]))
|
||||||
|
else:
|
||||||
|
# add code for drums samples
|
||||||
|
pass
|
||||||
|
return all_beats
|
||||||
|
|
||||||
|
def to_midi(samples, output_path=settings.generated_midi_path, program=0, tempo=120, beat_resolution=settings.beat_resolution):
|
||||||
|
tracks = [roll.Track(samples, program=program)]
|
||||||
|
return_midi = roll.Multitrack(tracks=tracks, tempo=tempo, downbeat=[0, 96, 192, 288], beat_resolution=beat_resolution)
|
||||||
|
roll.write(return_midi, settings.generated_midi_path)
|
||||||
|
|
||||||
|
def delete_empty_samples(sample_pack):
|
||||||
|
print('Deleting empty samples...')
|
||||||
|
temp_sample_pack = sample_pack
|
||||||
|
index_manipulator = 1
|
||||||
|
for index, sample in enumerate(sample_pack):
|
||||||
|
if sample.sum() == 0:
|
||||||
|
temp_sample_pack = np.delete(temp_sample_pack, index-index_manipulator, axis=0)
|
||||||
|
index_manipulator = index_manipulator + 1
|
||||||
|
print('Deleted {} empty samples'.format(index_manipulator-1))
|
||||||
|
return temp_sample_pack
|
||||||
|
|
||||||
|
def main():
|
||||||
|
|
||||||
|
if sys.argv[1]=='export':
|
||||||
|
print('Exporting started...')
|
||||||
|
|
||||||
|
sample_pack = np.empty((0,settings.midi_resolution,128))
|
||||||
|
|
||||||
|
for midi_file in os.listdir(settings.midi_dir):
|
||||||
|
midi_file_path = '{}/{}'.format(settings.midi_dir, midi_file)
|
||||||
|
midi_samples = to_samples(midi_file_path)
|
||||||
|
if midi_samples is None:
|
||||||
|
continue
|
||||||
|
sample_pack = np.concatenate((midi_samples, sample_pack), axis=0)
|
||||||
|
|
||||||
|
# sample_pack = delete_empty_samples(sample_pack)
|
||||||
|
|
||||||
|
np.savez_compressed(settings.samples_dir, sample_pack)
|
||||||
|
print('Exported {} samples'.format(sample_pack.shape[0]))
|
||||||
|
|
||||||
|
fig, axes = plt.subplots(nrows=10, ncols=10, figsize=(20, 20))
|
||||||
|
for idx, ax in enumerate(axes.ravel()):
|
||||||
|
n = np.random.randint(0, sample_pack.shape[0])
|
||||||
|
sample = sample_pack[n]
|
||||||
|
ax.imshow(sample, cmap = plt.get_cmap('gray'))
|
||||||
|
plt.savefig(settings.sample_preview_path)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
||||||
@ -1,249 +0,0 @@
|
|||||||
import settings
|
|
||||||
import pypianoroll as roll
|
|
||||||
import matplotlib.pyplot as plt
|
|
||||||
import numpy as np
|
|
||||||
import os
|
|
||||||
from tqdm import tqdm
|
|
||||||
from math import floor
|
|
||||||
import sys
|
|
||||||
from sklearn.preprocessing import MinMaxScaler
|
|
||||||
|
|
||||||
midi_program = {
|
|
||||||
0 : 'Perc',
|
|
||||||
1 : 'Acoustic Grand Piano',
|
|
||||||
2 : 'Bright Acoustic Piano',
|
|
||||||
3 : 'Electric Grand Piano',
|
|
||||||
4 : 'Honky-tonk Piano',
|
|
||||||
5 : 'Electric Piano 1',
|
|
||||||
6 : 'Electric Piano 2',
|
|
||||||
7 : 'Harpsichord',
|
|
||||||
8 : 'Clavi',
|
|
||||||
9 : 'Celesta',
|
|
||||||
10 : 'Glockenspiel',
|
|
||||||
11 : 'Music Box',
|
|
||||||
12 : 'Vibraphone',
|
|
||||||
13 : 'Marimba',
|
|
||||||
14 : 'Xylophone',
|
|
||||||
15 : 'Tubular Bells',
|
|
||||||
16 : 'Dulcimer',
|
|
||||||
17 : 'Drawbar Organ',
|
|
||||||
18 : 'Percussive Organ',
|
|
||||||
19 : 'Rock Organ',
|
|
||||||
20 : 'Church Organ',
|
|
||||||
21 : 'Reed Organ',
|
|
||||||
22 : 'Accordion',
|
|
||||||
23 : 'Harmonica',
|
|
||||||
24 : 'Tango Accordion',
|
|
||||||
25 : 'Acoustic Guitar (nylon)',
|
|
||||||
26 : 'Acoustic Guitar (steel)',
|
|
||||||
27 : 'Electric Guitar (jazz)',
|
|
||||||
28 : 'Electric Guitar (clean)',
|
|
||||||
29 : 'Electric Guitar (muted)',
|
|
||||||
30 : 'Overdriven Guitar',
|
|
||||||
31 : 'Distortion Guitar',
|
|
||||||
32 : 'Guitar harmonics',
|
|
||||||
33 : 'Acoustic Bass',
|
|
||||||
34 : 'Electric Bass (finger)',
|
|
||||||
35 : 'Electric Bass (pick)',
|
|
||||||
36 : 'Fretless Bass',
|
|
||||||
37 : 'Slap Bass 1',
|
|
||||||
38 : 'Slap Bass 2',
|
|
||||||
39 : 'Synth Bass 1',
|
|
||||||
40 : 'Synth Bass 2',
|
|
||||||
41 : 'Violin',
|
|
||||||
42 : 'Viola',
|
|
||||||
43 : 'Cello',
|
|
||||||
44 : 'Contrabass',
|
|
||||||
45 : 'Tremolo Strings',
|
|
||||||
46 : 'Pizzicato Strings',
|
|
||||||
47 : 'Orchestral Harp',
|
|
||||||
48 : 'Timpani',
|
|
||||||
49 : 'String Ensemble 1',
|
|
||||||
50 : 'String Ensemble 2',
|
|
||||||
51 : 'SynthStrings 1',
|
|
||||||
52 : 'SynthStrings 2',
|
|
||||||
53 : 'Choir Aahs',
|
|
||||||
54 : 'Voice Oohs',
|
|
||||||
55 : 'Synth Voice',
|
|
||||||
56 : 'Orchestra Hit',
|
|
||||||
57 : 'Trumpet',
|
|
||||||
58 : 'Trombone',
|
|
||||||
59 : 'Tuba',
|
|
||||||
60 : 'Muted Trumpet',
|
|
||||||
61 : 'French Horn',
|
|
||||||
62 : 'Brass Section',
|
|
||||||
63 : 'SynthBrass 1',
|
|
||||||
64 : 'SynthBrass 2',
|
|
||||||
65 : 'Soprano Sax',
|
|
||||||
66 : 'Alto Sax',
|
|
||||||
67 : 'Tenor Sax',
|
|
||||||
68 : 'Baritone Sax',
|
|
||||||
69 : 'Oboe',
|
|
||||||
70 : 'English Horn',
|
|
||||||
71 : 'Bassoon',
|
|
||||||
72 : 'Clarinet',
|
|
||||||
73 : 'Piccolo',
|
|
||||||
74 : 'Flute',
|
|
||||||
75 : 'Recorder',
|
|
||||||
76 : 'Pan Flute',
|
|
||||||
77 : 'Blown Bottle',
|
|
||||||
78 : 'Shakuhachi',
|
|
||||||
79 : 'Whistle',
|
|
||||||
80 : 'Ocarina',
|
|
||||||
81 : 'Lead 1 (square)',
|
|
||||||
82 : 'Lead 2 (sawtooth)',
|
|
||||||
83 : 'Lead 3 (calliope)',
|
|
||||||
84 : 'Lead 4 (chiff)',
|
|
||||||
85 : 'Lead 5 (charang)',
|
|
||||||
86 : 'Lead 6 (voice)',
|
|
||||||
87 : 'Lead 7 (fifths)',
|
|
||||||
88 : 'Lead 8 (bass + lead)',
|
|
||||||
89 : 'Pad 1 (new age)',
|
|
||||||
90 : 'Pad 2 (warm)',
|
|
||||||
91 : 'Pad 3 (polysynth)',
|
|
||||||
92 : 'Pad 4 (choir)',
|
|
||||||
93 : 'Pad 5 (bowed)',
|
|
||||||
94 : 'Pad 6 (metallic)',
|
|
||||||
95 : 'Pad 7 (halo)',
|
|
||||||
96 : 'Pad 8 (sweep)',
|
|
||||||
97 : 'FX 1 (rain)',
|
|
||||||
98 : 'FX 2 (soundtrack)',
|
|
||||||
99 : 'FX 3 (crystal)',
|
|
||||||
100 : 'FX 4 (atmosphere)',
|
|
||||||
101 : 'FX 5 (brightness)',
|
|
||||||
102 : 'FX 6 (goblins)',
|
|
||||||
103 : 'FX 7 (echoes)',
|
|
||||||
104 : 'FX 8 (sci-fi)',
|
|
||||||
105 : 'Sitar',
|
|
||||||
106 : 'Banjo',
|
|
||||||
107 : 'Shamisen',
|
|
||||||
108 : 'Koto',
|
|
||||||
109 : 'Kalimba',
|
|
||||||
110 : 'Bag pipe',
|
|
||||||
111 : 'Fiddle',
|
|
||||||
112 : 'Shanai',
|
|
||||||
113 : 'Tinkle Bell',
|
|
||||||
114 : 'Agogo',
|
|
||||||
115 : 'Steel Drums',
|
|
||||||
116 : 'Woodblock',
|
|
||||||
117 : 'Taiko Drum',
|
|
||||||
118 : 'Melodic Tom',
|
|
||||||
119 : 'Synth Drum',
|
|
||||||
120 : 'Reverse Cymbal',
|
|
||||||
121 : 'Guitar Fret Noise',
|
|
||||||
122 : 'Breath Noise',
|
|
||||||
123 : 'Seashore',
|
|
||||||
124 : 'Bird Tweet',
|
|
||||||
125 : 'Telephone Ring',
|
|
||||||
126 : 'Helicopter',
|
|
||||||
127 : 'Applause',
|
|
||||||
128 : 'Gunshot'
|
|
||||||
}
|
|
||||||
|
|
||||||
# strasznie wolna funcja ;/
|
|
||||||
def trim_notes(pianoroll):
|
|
||||||
now_block = []
|
|
||||||
for x in pianoroll:
|
|
||||||
this = None
|
|
||||||
prev = None
|
|
||||||
new_line =[]
|
|
||||||
for y in x:
|
|
||||||
this = y
|
|
||||||
if prev != None:
|
|
||||||
if this > 0 and prev > 0:
|
|
||||||
new_line.append(0)
|
|
||||||
else:
|
|
||||||
new_line.append(y)
|
|
||||||
else:
|
|
||||||
new_line.append(y)
|
|
||||||
prev = this
|
|
||||||
now_block.append(new_line)
|
|
||||||
return np.array(now_block)
|
|
||||||
|
|
||||||
def metrum_check(midi_lenght, metrum=4, beat_resolution=24):
|
|
||||||
return True if midi_lenght % (metrum * beat_resolution) == 0 else False
|
|
||||||
|
|
||||||
# >>> data = [[-1, 2], [-0.5, 6], [0, 10], [1, 18]]
|
|
||||||
# >>> scaler = MinMaxScaler()
|
|
||||||
# >>> print(scaler.fit(data))
|
|
||||||
# MinMaxScaler(copy=True, feature_range=(0, 1))
|
|
||||||
# >>> print(scaler.data_max_)
|
|
||||||
# [ 1. 18.]
|
|
||||||
# >>> print(scaler.transform(data))
|
|
||||||
# [[0. 0. ]
|
|
||||||
# [0.25 0.25]
|
|
||||||
# [0.5 0.5 ]
|
|
||||||
# [1. 1. ]]
|
|
||||||
# >>> print(scaler.transform([[2, 2]]))
|
|
||||||
# [[1.5 0. ]]
|
|
||||||
|
|
||||||
|
|
||||||
def to_samples(midi_file_path, midi_res=settings.midi_resolution, ignore_note_lenght=settings.ignore_note_lenght):
|
|
||||||
print('exporting samples from: {}'.format(midi_file_path))
|
|
||||||
midi_file = roll.Multitrack(midi_file_path)
|
|
||||||
samples = None
|
|
||||||
all_samples = np.empty((0,settings.midi_resolution,128))
|
|
||||||
for track in midi_file.tracks:
|
|
||||||
# if not track.is_drum:
|
|
||||||
if not metrum_check(track.pianoroll.shape[0]):
|
|
||||||
print('Track skipped')
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
instrument_track = track.pianoroll
|
|
||||||
instrument_track = trim_notes(instrument_track.T).T if ignore_note_lenght else instrument_track
|
|
||||||
scaler = MinMaxScaler()
|
|
||||||
instrument_track = scaler.fit_transform(instrument_track)
|
|
||||||
whole_beats = int(instrument_track.shape[0] / midi_res)
|
|
||||||
samples = instrument_track.reshape(whole_beats, midi_res, 128)
|
|
||||||
print('Exported {} samples of {}'.format(whole_beats, midi_program[track.program]))
|
|
||||||
all_samples = np.concatenate([samples, all_samples], axis=0)
|
|
||||||
return all_samples
|
|
||||||
|
|
||||||
def to_midi(samples, output_path=settings.generated_midi_path, program=0, tempo=120, beat_resolution=settings.beat_resolution):
|
|
||||||
tracks = [roll.Track(samples, program=program)]
|
|
||||||
return_midi = roll.Multitrack(tracks=tracks, tempo=tempo, downbeat=[0, 96, 192, 288], beat_resolution=beat_resolution)
|
|
||||||
roll.write(return_midi, settings.generated_midi_path)
|
|
||||||
|
|
||||||
def to_png(samples, output_path=settings.generated_pianoroll_path, horizontal=True):
|
|
||||||
img = samples.T if horizontal else samples
|
|
||||||
plt.imshow(img, cmap='gray')
|
|
||||||
plt.savefig(output_path)
|
|
||||||
|
|
||||||
def delete_empty_samples(sample_pack):
|
|
||||||
temp_sample_pack = sample_pack
|
|
||||||
index_manipulator = 1
|
|
||||||
for index, sample in enumerate(sample_pack):
|
|
||||||
if sample.sum() == 0:
|
|
||||||
temp_sample_pack = np.delete(temp_sample_pack, index-index_manipulator, axis=0)
|
|
||||||
index_manipulator = index_manipulator + 1
|
|
||||||
print('Deleted {} empty samples'.format(index_manipulator-1))
|
|
||||||
return temp_sample_pack
|
|
||||||
|
|
||||||
def main():
|
|
||||||
if sys.argv[1]=='-e':
|
|
||||||
print('Exporting started...')
|
|
||||||
sample_pack = np.empty((0,settings.midi_resolution,128))
|
|
||||||
for midi_file in os.listdir(settings.midi_dir):
|
|
||||||
midi_file_path = '{}/{}'.format(settings.midi_dir, midi_file)
|
|
||||||
midi_samples = to_samples(midi_file_path)
|
|
||||||
if midi_samples is None:
|
|
||||||
continue
|
|
||||||
sample_pack = np.concatenate((midi_samples, sample_pack), axis=0)
|
|
||||||
|
|
||||||
sample_pack = delete_empty_samples(sample_pack)
|
|
||||||
np.savez_compressed(settings.samples_dir, sample_pack)
|
|
||||||
print('Exported {} samples'.format(sample_pack.shape[0]))
|
|
||||||
elif sys.argv[1]=='-c':
|
|
||||||
to_midi(settings.generated_sample_path)
|
|
||||||
print('Samples to midi saved to {}'.format(settings.generated_sample_path))
|
|
||||||
elif sys.argv[1]=='-p':
|
|
||||||
sample_pack = np.load(settings.samples_path)['arr_0']
|
|
||||||
for i, sample in tqdm(enumerate(sample_pack)):
|
|
||||||
to_png(sample, output_path='data/preview/{}.png'.format(i))
|
|
||||||
if i>50:
|
|
||||||
sys.exit()
|
|
||||||
else:
|
|
||||||
print('type command afrer -e to export samples, -c to convert samples to midi, -p to preview samples in png')
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
main()
|
|
||||||
Loading…
Reference in New Issue
Block a user