File size: 1,325 Bytes
151b875 |
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 |
from argparse import ArgumentParser
from pathlib import Path
from anticipation.vocab import AUTOREGRESS, ANTICIPATE
from anticipation.convert import events_to_midi, interarrival_to_midi
if __name__ == '__main__':
parser = ArgumentParser(description='auditory check for a tokenized dataset')
parser.add_argument('filename',
help='file containing a tokenized MIDI dataset')
parser.add_argument('index', type=int, default=0,
help='the item to examine')
parser.add_argument('range', type=int, default=1,
help='range of items to examine')
args = parser.parse_args()
with open(args.filename, 'r') as f:
for i, line in enumerate(f):
if i < args.index:
continue
if i == args.index+args.range:
break
tokens = [int(token) for token in line.split()]
if tokens[0] in [AUTOREGRESS, ANTICIPATE]:
tokens = tokens[1:] # strip control codes
mid = events_to_midi(tokens)
else: # it's the interarrival tokenization
mid = interarrival_to_midi(tokens)
mid.save(f'output/{Path(args.filename).stem}{i}.mid')
print(f'{i} Tokenized MIDI Length: {mid.length} seconds ({len(tokens)} tokens)')
|