diff --git a/samples/python/audio_spectrogram.py b/samples/python/audio_spectrogram.py index f211bbb584..cc156888d4 100644 --- a/samples/python/audio_spectrogram.py +++ b/samples/python/audio_spectrogram.py @@ -330,6 +330,10 @@ class AudioDrawing: """ time_step = self.windLen - self.overlap + if time_step <= 0: + raise ValueError( + "Invalid STFT parameters: overlap must be smaller than window length" + ) stft = [] if self.windowType == "Hann": @@ -801,4 +805,4 @@ if __name__ == "__main__": args = parser.parse_args() - AudioDrawing(args).Draw() + AudioDrawing(args).Draw() \ No newline at end of file