testmfccq15.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import cmsisdsp as dsp
  2. import numpy as np
  3. import cmsisdsp.fixedpoint as f
  4. import cmsisdsp.mfcc as mfcc
  5. import scipy.signal as sig
  6. from mfccdebugdata import *
  7. from cmsisdsp.datatype import Q15
  8. import cmsisdsp.datatype as dt
  9. mfccq15=dsp.arm_mfcc_instance_q15()
  10. sample_rate = 16000
  11. FFTSize = 256
  12. numOfDctOutputs = 13
  13. freq_min = 64
  14. freq_high = sample_rate / 2
  15. numOfMelFilters = 20
  16. windowQ15 = dt.convert(sig.hamming(FFTSize, sym=False),Q15)
  17. filtLen,filtPos,packedFiltersQ15 = mfcc.melFilterMatrix(Q15,freq_min, freq_high, numOfMelFilters,sample_rate,FFTSize)
  18. dctMatrixFiltersQ15 = mfcc.dctMatrix(Q15,numOfDctOutputs, numOfMelFilters)
  19. status=dsp.arm_mfcc_init_q15(mfccq15,FFTSize,numOfMelFilters,numOfDctOutputs,
  20. dctMatrixFiltersQ15,
  21. filtPos,filtLen,packedFiltersQ15,windowQ15)
  22. print("Init status = %d" % status)
  23. tmp=np.zeros(2*FFTSize,dtype=np.int32)
  24. debugQ15 = f.toQ15(debug)
  25. errorStatus,resQ15=dsp.arm_mfcc_q15(mfccq15,debugQ15,tmp)
  26. print("MFCC status = %d" % errorStatus)
  27. res=(1<<8)*f.Q15toF32(resQ15)
  28. print(res)
  29. print(ref)
  30. print("FFT Length = %d" % mfccq15.fftLen())
  31. print("Nb MEL Filters = %d" % mfccq15.nbMelFilters())
  32. print("Nb DCT Outputs = %d" % mfccq15.nbDctOutputs())