Spaces:
Sleeping
Sleeping
File size: 1,616 Bytes
91394e0 |
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 |
from .base import MelodyDatasetHandler
class KiSing(MelodyDatasetHandler):
name = "kising"
def __init__(self, melody_type, cache_dir, *args, **kwargs):
# melody_type: support alignment type for "sample" melody source
import json
from datasets import load_dataset
song_db = load_dataset(
"jhansss/kising_score_segments", cache_dir=cache_dir, split="train"
).to_pandas()
song_db.set_index("segment_id", inplace=True)
assert (
song_db.index.is_unique
), "KiSing score segments should have unique segment_id."
if melody_type == "lyric":
with open("data/kising/song2word_lengths.json", "r") as f:
song2word_lengths = json.load(f)
elif melody_type == "note":
with open("data/kising/song2note_lengths.json", "r") as f:
song2word_lengths = json.load(f)
self.song_db = song_db
self.song2word_lengths = song2word_lengths
def get_song_ids(self):
return list(self.song2word_lengths.keys())
def get_phrase_length(self, song_id):
return self.song2word_lengths[song_id]
def iter_song_phrases(self, song_id):
segment_id = 1
while f"{song_id}_{segment_id:03d}" in self.song_db.index:
segment = self.song_db.loc[f"{song_id}_{segment_id:03d}"].to_dict()
segment["note_lyrics"] = [
lyric.strip("<>") if lyric in ["<AP>", "<SP>"] else lyric
for lyric in segment["note_lyrics"]
]
yield segment
segment_id += 1
|