File size: 1,176 Bytes
8061544 | 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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | """
mai_parser - Maimai (舞萌) chart data parser for ML training.
Usage:
from mai_parser import SongDataset, parse_maidata_file
# Load entire dataset
ds = SongDataset("datasets").load()
print(ds.stats())
# Parse single file
song = parse_maidata_file("datasets/10/maidata.txt")
print(song.title_clean, song.artist, song.bpm)
# Convert chart to tensor for training
chart = song.get_chart(Difficulty.MASTER)
tensor = chart_to_tensor(chart)
"""
from .models import Cabinet, Chart, Difficulty, Song, TouchNote
from .parser import parse_maidata, parse_maidata_file
from .dataset import (
ChartDataset,
SongDataset,
chart_to_sequence,
chart_to_tensor,
note_to_vector,
NOTE_TYPE,
NOTE_TYPE_INV,
NUM_POSITIONS,
HAS_TORCH,
)
__all__ = [
# Models
"Cabinet",
"Chart",
"Difficulty",
"Song",
"TouchNote",
# Parser
"parse_maidata",
"parse_maidata_file",
# Dataset
"ChartDataset",
"SongDataset",
"chart_to_sequence",
"chart_to_tensor",
"note_to_vector",
# Constants
"NOTE_TYPE",
"NOTE_TYPE_INV",
"NUM_POSITIONS",
"HAS_TORCH",
]
|