From afba90f9b97384caa3f6e3af7f23faa4e8e4a019 Mon Sep 17 00:00:00 2001 From: Yuanchao Du Date: Fri, 30 Aug 2019 10:50:00 +0800 Subject: [PATCH] fix bug, change squre to element wise squre --- pyAudioAnalysis/audioFeatureExtraction.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pyAudioAnalysis/audioFeatureExtraction.py b/pyAudioAnalysis/audioFeatureExtraction.py index 4f159ff3f..f0d96f9e3 100644 --- a/pyAudioAnalysis/audioFeatureExtraction.py +++ b/pyAudioAnalysis/audioFeatureExtraction.py @@ -25,12 +25,12 @@ def stZCR(frame): def stEnergy(frame): """Computes signal energy of frame""" - return numpy.sum(frame ** 2) / numpy.float64(len(frame)) + return numpy.sum(np.square(frame, dtype='int64')) / numpy.float64(len(frame)) def stEnergyEntropy(frame, n_short_blocks=10): """Computes entropy of energy""" - Eol = numpy.sum(frame ** 2) # total frame energy + Eol = numpy.sum(np.square(frame, dtype='int64')) # total frame energy L = len(frame) sub_win_len = int(numpy.floor(L / n_short_blocks)) if L != sub_win_len * n_short_blocks: @@ -139,7 +139,7 @@ def stHarmonic(frame, fs): M = len(R) - 1 Gamma = numpy.zeros((M), dtype=numpy.float64) - CSum = numpy.cumsum(frame ** 2) + CSum = numpy.cumsum(np.square(frame, dtype='int64')) Gamma[m0:M] = R[m0:M] / (numpy.sqrt((g * CSum[M:m0:-1])) + eps) ZCR = stZCR(Gamma)