fix: Chroma rolling半個音(#42)

Keycatowo · May 16, 2023 · 40f1c29 · 40f1c29
1 parent 66a2f35
commit 40f1c29
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/pitch_estimation.py b/src/pitch_estimation.py
@@ -201,6 +201,8 @@ def plot_pitch_class(
     )
     note_probs = np.mean(chroma >= 1/np.sqrt(2), axis=1)
     note_probs = note_probs / note_probs.sum() # normalize
+    # 將note_probs rolling (resolution_ratio/2)格
+    note_probs = np.roll(note_probs, resolution_ratio//2)
 
     # 建立位置mask
     unit_pos = np.zeros(resolution_ratio, dtype=bool) # 一個音的標記長度