ref: 89bc6f4938d3c1e41e351d7cd8314ed3fb3cd433
dir: /python/demos/demo_reading_speed.py/
#! /usr/bin/env python # -*- coding: utf-8 -*- """ Compare the speed of several methods for reading and loading a sound file. Optionally, this file can make use of the following packages: - audioread https://github.com/beetbox/audioread - scipy https://scipy.org - librosa https://github.com/bmcfee/librosa - pydub https://github.com/jiaaro/pydub Uncomment the function names below and send us your speed results! """ test_functions = [ "read_file_aubio", "load_file_aubio", #"load_file_scipy", #"load_file_scipy_mmap", #"read_file_audioread", #"load_file_librosa", #"read_file_pydub", #"load_file_pydub", ] import numpy as np def read_file_audioread(filename): import audioread # taken from librosa.util.utils def convert_buffer_to_float(buf, n_bytes = 2, dtype = np.float32): # Invert the scale of the data scale = 1./float(1 << ((8 * n_bytes) - 1)) # Construct the format string fmt = '<i{:d}'.format(n_bytes) # Rescale and format the data buffer out = scale * np.frombuffer(buf, fmt).astype(dtype) return out with audioread.audio_open(filename) as f: total_frames = 0 for buf in f: samples = convert_buffer_to_float(buf) samples = samples.reshape(f.channels, -1) total_frames += samples.shape[1] return total_frames, f.samplerate def load_file_librosa(filename): import librosa y, sr = librosa.load(filename, sr = None) #print y.mean(), y.shape return len(y), sr def load_file_scipy(filename): import scipy.io.wavfile sr, y = scipy.io.wavfile.read(filename) y = y.astype('float32') / 32767 #print y.mean(), y.shape return len(y), sr def load_file_scipy_mmap(filename): import scipy.io.wavfile sr, y = scipy.io.wavfile.read(filename, mmap = True) #print y.mean(), y.shape return len(y), sr def read_file_pydub(filename): from pydub import AudioSegment song = AudioSegment.from_file(filename) song.get_array_of_samples() return song.frame_count(), song.frame_rate def load_file_pydub(filename): from pydub import AudioSegment song = AudioSegment.from_file(filename) y = np.asarray(song.get_array_of_samples(), dtype = 'float32') y = y.reshape(song.channels, -1) / 32767. return song.frame_count(), song.frame_rate def read_file_aubio(filename): import aubio f = aubio.source(filename, hop_size = 1024) total_frames = 0 while True: _, read = f() total_frames += read if read < f.hop_size: break return total_frames, f.samplerate def load_file_aubio(filename): import aubio f = aubio.source(filename, hop_size = 1024) y = np.zeros(f.duration, dtype = aubio.float_type) total_frames = 0 while True: samples, read = f() y[total_frames:total_frames + read] = samples[:read] total_frames += read if read < f.hop_size: break assert len(y) == total_frames #print y.mean(), y.shape return total_frames, f.samplerate def test_speed(function, filename): times = [] for _ in range(10): start = time.time() try: total_frames, samplerate = function(filename) except ImportError as e: print ("error: failed importing {:s}".format(e)) return elapsed = time.time() - start #print ("{:5f} ".format(elapsed)), times.append(elapsed) #print times = np.array(times) duration_min = int(total_frames/float(samplerate) // 60) str_format = '{:25s} took {:5f} seconds avg (±{:5f}) to run on a ~ {:d} minutes long file' print (str_format.format(function.__name__, times.mean(), times.std(), duration_min )) if __name__ == '__main__': import sys, time if len(sys.argv) < 2: print ("not enough arguments") sys.exit(1) filename = sys.argv[1] for f in test_functions: # get actual function from globals test_function = globals()[f] test_speed(test_function, filename)