data extraction to groups of instruments; drums is now avaible
This commit is contained in:
parent
b296eb9e6c
commit
dac8add955
@ -30,7 +30,9 @@ def to_samples(midi_file_path, midi_res=settings.midi_resolution):
|
||||
|
||||
for track in roll.Multitrack(midi_file_path).tracks:
|
||||
if not track.is_drum:
|
||||
key = track.program + 1
|
||||
key = settings.midi_group[track.program + 1]
|
||||
else:
|
||||
key = 'Drums'
|
||||
|
||||
# this makes pack of samples of N x 96 x 128 shape
|
||||
number_of_beats = floor(track.pianoroll.shape[0] / midi_res)
|
||||
@ -38,14 +40,12 @@ def to_samples(midi_file_path, midi_res=settings.midi_resolution):
|
||||
track_beats = track_pianoroll.reshape(number_of_beats, midi_res, 128)
|
||||
|
||||
# save collected pack of data to dictionary with samples packs for every instrument
|
||||
samples_by_instrument[track.program + 1] = np.concatenate([track_beats, samples_by_instrument[ track.program + 1]], axis=0)
|
||||
else:
|
||||
# TODO: add code for drums samples
|
||||
pass
|
||||
samples_by_instrument[key] = np.concatenate([track_beats, samples_by_instrument[key]], axis=0)
|
||||
|
||||
return samples_by_instrument
|
||||
|
||||
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)]
|
||||
def to_midi(samples, output_path=settings.generated_midi_path, program=0, tempo=120, is_drum=False, beat_resolution=settings.beat_resolution):
|
||||
tracks = [roll.Track(samples, program=program, is_drum=is_drum)]
|
||||
return_midi = roll.Multitrack(tracks=tracks, tempo=tempo, downbeat=[0, 96, 192, 288], beat_resolution=beat_resolution)
|
||||
roll.write(return_midi, output_path)
|
||||
return return_midi
|
||||
@ -72,6 +72,7 @@ def main():
|
||||
pass
|
||||
if midi_samples is None:
|
||||
continue
|
||||
|
||||
# this is for intrument separation
|
||||
for key, value in midi_samples.items():
|
||||
value = delete_empty_samples(value)
|
||||
@ -86,7 +87,7 @@ def main():
|
||||
for key, value in tqdm(samples_pack_by_instrument.items()):
|
||||
if not os.path.exists(output_path):
|
||||
os.makedirs(output_path)
|
||||
np.savez_compressed('{}/{}.npz'.format(output_path, settings.midi_program[key]), value)
|
||||
np.savez_compressed('{}/{}.npz'.format(output_path, key), value)
|
||||
|
||||
# # Give a preview of what samples looks like
|
||||
# fig, axes = plt.subplots(nrows=10, ncols=10, figsize=(20, 20))
|
||||
|
||||
@ -164,3 +164,150 @@ midi_program = {
|
||||
127 : 'Applause',
|
||||
128 : 'Gunshot'
|
||||
}
|
||||
|
||||
midi_group = {
|
||||
# Piano
|
||||
1 : 'Piano',
|
||||
2 : 'Piano',
|
||||
3 : 'Piano',
|
||||
4 : 'Piano',
|
||||
5 : 'Piano',
|
||||
6 : 'Piano',
|
||||
7 : 'Piano',
|
||||
8 : 'Piano',
|
||||
# Chromatic Percussion
|
||||
9 : 'Chromatic_Percussion',
|
||||
10 : 'Chromatic_Percussion',
|
||||
11 : 'Chromatic_Percussion',
|
||||
12 : 'Chromatic_Percussion',
|
||||
13 : 'Chromatic_Percussion',
|
||||
14 : 'Chromatic_Percussion',
|
||||
15 : 'Chromatic_Percussion',
|
||||
16 : 'Chromatic_Percussion',
|
||||
# Organ
|
||||
17 : 'Organ',
|
||||
18 : 'Organ',
|
||||
19 : 'Organ',
|
||||
20 : 'Organ',
|
||||
21 : 'Organ',
|
||||
22 : 'Organ',
|
||||
23 : 'Organ',
|
||||
24 : 'Organ',
|
||||
# Guitar
|
||||
25 : 'Guitar',
|
||||
26 : 'Guitar',
|
||||
27 : 'Guitar',
|
||||
28 : 'Guitar',
|
||||
29 : 'Guitar',
|
||||
30 : 'Guitar',
|
||||
31 : 'Guitar',
|
||||
32 : 'Guitar',
|
||||
# Bass
|
||||
33 : 'Bass',
|
||||
34 : 'Bass',
|
||||
35 : 'Bass',
|
||||
36 : 'Bass',
|
||||
37 : 'Bass',
|
||||
38 : 'Bass',
|
||||
39 : 'Bass',
|
||||
40 : 'Bass',
|
||||
# Strings
|
||||
41 : 'Strings',
|
||||
42 : 'Strings',
|
||||
43 : 'Strings',
|
||||
44 : 'Strings',
|
||||
45 : 'Strings',
|
||||
46 : 'Strings',
|
||||
47 : 'Strings',
|
||||
48 : 'Strings',
|
||||
# Ensemble
|
||||
49 : 'Ensemble',
|
||||
50 : 'Ensemble',
|
||||
51 : 'Ensemble',
|
||||
52 : 'Ensemble',
|
||||
53 : 'Ensemble',
|
||||
54 : 'Ensemble',
|
||||
55 : 'Ensemblee',
|
||||
56 : 'Ensemble',
|
||||
# Brass
|
||||
57 : 'Brass',
|
||||
58 : 'Brass',
|
||||
59 : 'Brass',
|
||||
60 : 'Brass',
|
||||
61 : 'Brass',
|
||||
62 : 'Brass',
|
||||
63 : 'Brass',
|
||||
64 : 'Brass',
|
||||
# Reed
|
||||
65 : 'Reed',
|
||||
66 : 'Reed',
|
||||
67 : 'Reed',
|
||||
68 : 'Reed',
|
||||
69 : 'Reed',
|
||||
70 : 'Reed',
|
||||
71 : 'Reed',
|
||||
72 : 'Reed',
|
||||
# Pipe
|
||||
73 : 'Pipe',
|
||||
74 : 'Pipe',
|
||||
75 : 'Pipe',
|
||||
76 : 'Pipe',
|
||||
77 : 'Pipe',
|
||||
78 : 'Pipe',
|
||||
79 : 'Pipe',
|
||||
80 : 'Pipe',
|
||||
# Synth Lead
|
||||
81 : 'Synth_Lead',
|
||||
82 : 'Synth_Lead',
|
||||
83 : 'Synth_Lead',
|
||||
84 : 'Synth_Lead',
|
||||
85 : 'Synth_Lead',
|
||||
86 : 'Synth_Lead',
|
||||
87 : 'Synth_Lead',
|
||||
88 : 'Synth_Lead',
|
||||
# Synth Pad
|
||||
89 : 'Synth_Pad',
|
||||
90 : 'Synth_Pad',
|
||||
91 : 'Synth_Pad',
|
||||
92 : 'Synth_Pad',
|
||||
93 : 'Synth_Pad',
|
||||
94 : 'Synth_Pad',
|
||||
95 : 'Synth_Pad',
|
||||
96 : 'Synth_Pad',
|
||||
# Synth Effects
|
||||
97 : 'Synth_Effects',
|
||||
98 : 'Synth_Effects',
|
||||
99 : 'Synth_Effects',
|
||||
100 : 'Synth_Effects',
|
||||
101 : 'Synth_Effects',
|
||||
102 : 'Synth_Effects',
|
||||
103 : 'Synth_Effects',
|
||||
104 : 'Synth_Effects',
|
||||
# Ethnic
|
||||
105 : 'Ethnic',
|
||||
106 : 'Ethnic',
|
||||
107 : 'Ethnic',
|
||||
108 : 'Ethnic',
|
||||
109 : 'Ethnic',
|
||||
110 : 'Ethnic',
|
||||
111 : 'Ethnic',
|
||||
112 : 'Ethnic',
|
||||
# Percussive
|
||||
113 : 'Percussive',
|
||||
114 : 'Percussive',
|
||||
115 : 'Percussive',
|
||||
116 : 'Percussive',
|
||||
117 : 'Percussive',
|
||||
118 : 'Percussive',
|
||||
119 : 'Percussive',
|
||||
120 : 'Percussive',
|
||||
# Sound Effects
|
||||
121 : 'Sound_Effects',
|
||||
122 : 'Sound_Effects',
|
||||
123 : 'Sound_Effects',
|
||||
124 : 'Sound_Effects',
|
||||
125 : 'Sound_Effects',
|
||||
126 : 'Sound_Effects',
|
||||
127 : 'Sound_Effects',
|
||||
128 : 'Sound_Effects'
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user