Update hz2midi to avoid log2(0) warnings

Behavior should remain unchanged
justinsalamon · Apr 23, 2018 · 4821cde · 4821cde
1 parent a4039f8
commit 4821cde
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/audio_to_midi_melodia.py b/audio_to_midi_melodia.py
@@ -138,9 +138,10 @@ def hz2midi(hz):
 
     # convert from Hz to midi note
     hz_nonneg = hz.copy()
-    hz_nonneg[hz <= 0] = 0
+    idx = hz_nonneg <= 0
+    hz_nonneg[idx] = 1
     midi = 69 + 12*np.log2(hz_nonneg/440.)
-    midi[midi <= 0] = 0
+    midi[idx] = 0
 
     # round
     midi = np.round(midi)