From cc5ba9fa4680a74dc84e31bf2e1cf906634bf2e4 Mon Sep 17 00:00:00 2001
From: ferrari <maxence.ferrari@gmail.com>
Date: Thu, 25 Mar 2021 16:52:10 +0100
Subject: [PATCH] Fix len song

---
 ipi_extract.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ipi_extract.py b/ipi_extract.py
index 49b5a06..33366cd 100644
--- a/ipi_extract.py
+++ b/ipi_extract.py
@@ -39,6 +39,8 @@ def load_file(in_path, channel, low, high):
     song = song[:, channel]
     sos = sg.butter(3, [low, high], 'bandpass', fs=sr, output='sos')
     song = sg.sosfiltfilt(sos, song)
+    if len(song) < 20*sr:
+        song = np.pad(song, (1, 20*sr), mode='constant')
     frac = Fraction(FSSR, sr)
     song_resample = sg.resample_poly(song, frac.numerator, frac.denominator)
     print('Done processing')
-- 
GitLab