complete example, REALLY make Python 2/3 compatible

scienceopen · ybdarrenwang · commit 07720bdde53a · 2016-08-14T17:26:03.000-07:00
diff --git a/python_speech_features/base.py b/python_speech_features/base.py
@@ -148,6 +148,14 @@ def get_filterbanks(nfilt=20,nfft=512,samplerate=16000,lowfreq=0,highfreq=None):
     #  from Hz to fft bin number
     bin = numpy.floor((nfft+1)*mel2hz(melpoints)/samplerate)
 
+    fbank = numpy.zeros([nfilt,nfft//2+1])
+    for j in range(0,nfilt):
+        for i in range(int(bin[j]), int(bin[j+1])):
+            fbank[j,i] = (i - bin[j]) / (bin[j+1]-bin[j])
+        for i in range(int(bin[j+1]), int(bin[j+2])):
+            fbank[j,i] = (bin[j+2]-i) / (bin[j+2]-bin[j+1])
+    return fbank                 
+    
 def lifter(cepstra, L=22):
     """Apply a cepstral lifter the the matrix of cepstra. This has the effect of increasing the
     magnitude of the high frequency DCT coeffs.
