bugfix for small files

LimitlessGreen · Aug 20, 2020 · f37f2c5 · f37f2c5
1 parent d32ca09
commit f37f2c5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/WaveletSegment.py b/WaveletSegment.py
@@ -642,8 +642,11 @@ def detectCalls(self, wf, nodelist, subfilter, rf=True, annotation=None, window=
             # nw=int(np.ceil(N / inc_sr))
             # detected = np.zeros(nw)
 
-            # Compute the energy curve (a la Jinnai et al. 2012)
-            E = ce.EnergyCurve(C, M)
+            if len(C) > 2*M+1:
+                # Compute the energy curve (a la Jinnai et al. 2012)
+                E = ce.EnergyCurve(C, M)
+            else:
+                break
             # Compute threshold using mean & sd from non-call sections
             # Virginia: changed the base. I'm using resol_sr as a base. Cause I'm looking for detections on windows.
             #This step is not so clear for me