File size: 892 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 |
from argparse import ArgumentParser
from anticipation.vocab import SEPARATOR
if __name__ == '__main__':
parser = ArgumentParser(description='inspect a MIDI dataset')
parser.add_argument('filename',
help='file containing a tokenized MIDI dataset')
parser.add_argument('index', type=int, default=0,
help='start index of items to examine')
parser.add_argument('range', type=int, default=1,
help='number 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+args.range:
break
if i >= args.index:
tokens = [int(token) for token in line.split()]
if SEPARATOR in tokens[1:]:
print(f'Sequence boundary in line {i}. Control codes {tokens[:1]}')
|