|
|
|
|
from os import listdir
|
|
|
|
|
from os.path import isfile, join
|
|
|
|
|
import multiprocessing as mp
|
|
|
|
|
import pandas as pd
|
|
|
|
|
from feature_extractor import extractFeatures
|
|
|
|
|
|
|
|
|
|
class bcolors:
|
|
|
|
|
BLUE = '\033[94m'
|
|
|
|
|
GREEN = '\033[92m'
|
|
|
|
|
YELLOW = '\033[93m'
|
|
|
|
|
RED = '\033[91m'
|
|
|
|
|
ENDC = '\033[0m'
|
|
|
|
|
|
|
|
|
|
def batchExtract(audioFilesPath, featureFilesPath, sampleRate):
|
|
|
|
|
audioFiles = [file for file in listdir(audioFilesPath) if isfile(join(audioFilesPath, file))]
|
|
|
|
|
|
|
|
|
|
dataframesList = [None]*len(audioFiles)
|
|
|
|
|
|
|
|
|
|
pool = mp.Pool()
|
|
|
|
|
for process, file in enumerate(audioFiles):
|
|
|
|
|
dataframesList[process] = pool.apply_async(extractFeatures,args=(audioFilesPath + file,
|
|
|
|
|
featureFilesPath + file[0:file.rfind('.')] + '.json',int(sampleRate))).get()
|
|
|
|
|
pool.close()
|
|
|
|
|
pool.join()
|
|
|
|
|
|
|
|
|
|
joinedDataset = pd.concat(dataframesList)
|
|
|
|
|
|
|
|
|
|
print('Batch feature extraction finished successfully.')
|
|
|
|
|
|
|
|
|
|
return joinedDataset
|
|
|
|
|
|
|
|
|
|
# Prints a nice message to let the user know the module was imported
|
|
|
|
|
print(bcolors.BLUE + 'batch_feature_extractor loaded' + bcolors.ENDC)
|
|
|
|
|
|
|
|
|
|
# Enables executing the module as a standalone script
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
import sys
|
|
|
|
|
batchExtract(sys.argv[1], sys.argv[2], sys.argv[3])
|
