Changes to be committed: 데이터셋 추가, 토크나이저 수정

new file: data/29.mid
new file: data/30.mid
modified: utillity/miditokenizer.py
deleted: utillity/test.py

utillity/miditokenizer.py

@@ -1,16 +1,21 @@
-from miditok import REMI, TokenizerConfig, TokSequence
+from miditok import REMI, TokSequence, TokenizerConfig
 from symusic import Score
+import os
 
 # 토크나이저 설정
 config = TokenizerConfig()
 tokenizer = REMI(config) # REMI 토크나이저 사용
 
-def tokenize_midi(midi_num: int):
-    midi = Score(f"data/{midi_num}.mid")
-    tokens = tokenizer(midi)
+def get_midi_scores(midi_folder):
+    for filename in os.listdir(midi_folder):
+        if filename.endswith(".mid"):
+            yield Score(os.path.join(midi_folder, filename)) # 제네레이터로 코루틴 만들어서 이용
 
-    return tokens
+def tokenize_midi(midi):
+    tokens = tokenizer(midi)
+    return tokens[0].ids # type: ignore
 
-def detokenize_midi(tokens:TokSequence):
-    midi = tokenizer(tokens)
+def detokenize_midi(tokens: list[int]):
+    seq = TokSequence(ids=tokens) # type: ignore
+    midi = tokenizer.decode([seq])
     return midi

utillity/test.py

@@ -1,70 +0,0 @@
-import pretty_midi
-import random
-
-# 슬픈 분위기: A minor 스케일
-A_MINOR_SCALE = [69, 71, 72, 74, 76, 77, 79]  # A B C D E F G
-
-# 코드 진행 (A minor 중심)
-CHORD_PROGRESSION = ['Am', 'F', 'C', 'G']
-CHORD_DICT = {
-    'Am': [69, 72, 76],   # A C E
-    'F':  [65, 69, 72],   # F A C
-    'C':  [60, 64, 67],   # C E G
-    'G':  [67, 71, 74],   # G B D
-}
-
-# 설정값
-BPM = 90
-CHORD_DURATION = 1.0
-NOTE_RESOLUTION = 0.5
-VELOCITY_CHORD = 70
-VELOCITY_MELODY = 90
-
-# MIDI 객체 생성
-midi = pretty_midi.PrettyMIDI()
-chord_track = pretty_midi.Instrument(program=0)    # 피아노
-melody_track = pretty_midi.Instrument(program=40)  # 바이올린
-
-# 초기 시간
-current_time = 0.0
-
-# 코드 진행 순회
-for chord_name in CHORD_PROGRESSION:
-    chord_pitches = CHORD_DICT[chord_name]
-    
-    # 코드 입력
-    for pitch in chord_pitches:
-        chord_track.notes.append(pretty_midi.Note(
-            velocity=VELOCITY_CHORD,
-            pitch=pitch,
-            start=current_time,
-            end=current_time + CHORD_DURATION
-        ))
-    
-    # 멜로디 구성 (슬픈 느낌: 자연단음계에서 부드럽게 흐름)
-    time_cursor = current_time
-    previous_pitch = None
-
-    while time_cursor < current_time + CHORD_DURATION:
-        pitch = random.choice(A_MINOR_SCALE)
-        
-        # 음 높이 흐름 부드럽게
-        if previous_pitch and abs(pitch - previous_pitch) > 5:
-            pitch = previous_pitch + random.choice([-2, -1, 1, 2])
-        
-        melody_track.notes.append(pretty_midi.Note(
-            velocity=VELOCITY_MELODY,
-            pitch=pitch,
-            start=time_cursor,
-            end=time_cursor + NOTE_RESOLUTION
-        ))
-        previous_pitch = pitch
-        time_cursor += NOTE_RESOLUTION
-
-    # 다음 코드로 이동
-    current_time += CHORD_DURATION
-
-# 트랙 추가 및 MIDI 파일 저장
-midi.instruments.append(chord_track)
-midi.instruments.append(melody_track)
-midi.write('sad_ambient_melody.mid')