diff utils/SegUtil.py @ 12:c23658e8ae38

fp feature notebook
author mitian
date Mon, 25 May 2015 17:27:48 +0100
parents 56a2ca9359d0
children cc8ceb270e79
line wrap: on
line diff
--- a/utils/SegUtil.py	Mon May 18 17:43:48 2015 +0100
+++ b/utils/SegUtil.py	Mon May 25 17:27:48 2015 +0100
@@ -346,6 +346,22 @@
 	return feature_array		
 
 
+def getRolloff(data, tpower, filterbank, thresh=0.9):
+	nFrames = data.shape[0]
+	nFilters = len(filterbank)
+	rolloff = np.zeros(nFrames)
+	for i in xrange(nFrames):
+		rolloffE = thresh * tpower[i]
+		temp = 0.0
+		tempE = 0.0
+		for band in xrange(nFilters):
+			temp += data[i][band]
+			if temp > rolloffE: break
+		rolloff[i] = filterbank[nFilters-band-1]
+	
+	return rolloff
+
+
 def verifyPeaks(peak_canditates, dev_list):
 	'''Verify peaks from the 1st round detection by applying adaptive thresholding to the deviation list.'''