summaryrefslogtreecommitdiff
path: root/python/spectrum.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/spectrum.py')
-rw-r--r--python/spectrum.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/python/spectrum.py b/python/spectrum.py
index 514092b..267f368 100644
--- a/python/spectrum.py
+++ b/python/spectrum.py
@@ -13,8 +13,8 @@ FILENAME = sys.argv[1]
N_FFT = 2048
def read_audio_spectum(filename):
- fs, x = scipy.io.wavfile.read(filename)
- S = librosa.stft(numpy.transpose(x)[0], N_FFT)
+ x, fs = librosa.load(filename, 44100)
+ S = librosa.stft(x, N_FFT)
S = numpy.log1p(numpy.abs(S))
return S, fs