]> git.parisson.com Git - timeside.git/commitdiff
Analyzer/odf : fix NaN issue
authorThomas Fillon <thomas@parisson.com>
Wed, 19 Mar 2014 13:05:53 +0000 (14:05 +0100)
committerThomas Fillon <thomas@parisson.com>
Wed, 19 Mar 2014 13:05:53 +0000 (14:05 +0100)
timeside/analyzer/odf.py

index 3f157fcaae9fccd9cb562714d646b112a5f8a840..c30244541dffd8c9d334920e8675d810a79596f5 100644 (file)
@@ -104,7 +104,9 @@ class OnsetDetectionFunction(Analyzer):
 
         # Summation along the frequency axis
         odf_diff = S_diff.sum(axis=1)
-        odf_diff = odf_diff / np.median(odf_diff)  # Normalize
+        odf_median = np.median(odf_diff)
+        if odf_median:
+            odf_diff = odf_diff / odf_median  # Normalize
 
         odf = self.new_result(data_mode='value', time_mode='framewise')
         #odf.parameters = {'FFT_SIZE': self.FFT_SIZE}