Shuadissen
48 lines
2.2 KiB
Python
48 lines
2.2 KiB
Python
import Extract_Features as features
|
|
from subprocess import call
|
|
import os
|
|
import sys
|
|
import shlex
|
|
import argparse
|
|
|
|
|
|
def easy_call(command, debug_mode=False):
|
|
try:
|
|
#command = "time " + command
|
|
if debug_mode:
|
|
print >>sys.stderr, command
|
|
call(command, shell=True)
|
|
except Exception as exception:
|
|
print "Error: could not execute the following"
|
|
print ">>", command
|
|
print type(exception) # the exception instance
|
|
print exception.args # arguments stored in .args
|
|
exit(-1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# parse arguments
|
|
parser = argparse.ArgumentParser(description='Extract features for formants estimation.')
|
|
parser.add_argument('wav_file', default='', help="WAV audio filename (single vowel or an whole utternace)")
|
|
parser.add_argument('formants_file', default='', help="output formant text file")
|
|
parser.add_argument('--begin', help="beginning time in the WAV file", default=0.0, type=float)
|
|
parser.add_argument('--end', help="end time in the WAV file", default=-1.0, type=float)
|
|
args = parser.parse_args()
|
|
full_path = os.path.realpath(__file__)
|
|
if not os.path.exists(os.path.dirname(full_path)+'/Features/'):
|
|
os.makedirs(os.path.dirname(full_path)+'/Features/')
|
|
if not os.path.exists(os.path.dirname(full_path)+'/Predictions/'):
|
|
os.makedirs(os.path.dirname(full_path)+'/Predictions/')
|
|
|
|
if args.begin > 0.0 or args.end > 0.0:
|
|
Data = features.Create_features(args.wav_file, args.formants_file, args.begin, args.end)
|
|
ff = str(os.path.dirname(os.path.realpath(__file__))+'/Features/features_' + args.formants_file+'.txt')
|
|
pf = str(os.path.dirname(os.path.realpath(__file__))+'/Predictions/' +args.formants_file+'.csv')
|
|
easy_call("th load_estimation_model.lua " + ff + ' ' + pf)
|
|
else:
|
|
Data = features.Create_features(args.wav_file, args.formants_file)
|
|
ff = str(os.path.dirname(os.path.realpath(__file__))+'/Features/features_' + args.formants_file+'.txt')
|
|
pf = str(os.path.dirname(os.path.realpath(__file__))+'/Predictions/' +args.formants_file+'.csv')
|
|
easy_call("th load_tracking_model.lua " + ff + ' ' + pf)
|
|
|