write maximum_pcen

BirdVox · Aug 16, 2019 · f43bf50 · f43bf50
1 parent eecd4f6
commit f43bf50
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/birdvoxpaint/indices.py b/birdvoxpaint/indices.py
@@ -28,3 +28,10 @@ def maximum_flux(S):
     spectrogram = np.abs(S)**2
     flux = np.abs(np.diff(spectrogram, axis=1))
     return np.max(flux, axis=1)
+
+
+def maximum_pcen(S, **kwargs):
+    spectrogram = np.abs(S)**2
+    kwargs["return_zf"] = True
+    pcen, zf = librosa.pcen(spectrogram, **kwargs)
+    return np.max(pcen, axis=1), zf