adding limit support

This commit is contained in:
sg3510 2015-02-26 14:33:41 +00:00
parent 2c961cc43f
commit 3a9e3a4da7

View file

@ -44,6 +44,9 @@ def read(filename, limit=None):
except audioop.error: except audioop.error:
fs, _, audiofile = wavio.readwav(filename) fs, _, audiofile = wavio.readwav(filename)
if limit:
audiofile = audiofile[:limit * 1000]
audiofile = audiofile.T audiofile = audiofile.T
audiofile = audiofile.astype(np.int16) audiofile = audiofile.astype(np.int16)