diff --git a/bob/pad/face/extractor/FrequencySpectrum.py b/bob/pad/face/extractor/FrequencySpectrum.py index d22dcde00e76b29af54eca37cebd1201ccc79f27..059dd91a33298ade8b84d4ffcfc2e5861c7e5d8a 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