From a59bf0c70f20cb825b040a427dcfacae67a7791b Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Tue, 27 Feb 2018 16:49:40 +0100
Subject: [PATCH] [extractor] added sanity check on the input for
FrequencySpectrum features
---
bob/pad/face/extractor/FrequencySpectrum.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/bob/pad/face/extractor/FrequencySpectrum.py b/bob/pad/face/extractor/FrequencySpectrum.py
index d22dcde0..059dd91a 100644
--- a/bob/pad/face/extractor/FrequencySpectrum.py
+++ b/bob/pad/face/extractor/FrequencySpectrum.py
@@ -54,6 +54,14 @@ class FrequencySpectrum(Extractor, object):
freq: numpy.array
the frequency spectrum
"""
+ # sanity check
+ if signal.ndim == 1:
+ if numpy.isnan(numpy.sum(signal)):
+ return
+ if signal.ndim == 2 and (signal.shape[1] == 3):
+ if numpy.isnan(numpy.sum(signal[:, 1])):
+ return
+
output_dim = int((self.nfft / 2) + 1)
# we have a single pulse signal
--
GitLab