azalea/DeepFormants
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

return predictions

Browse Source
This commit is contained in:
Jason
2022-04-03 11:44:59 -07:00
parent 826f5bc486
commit da1667e241
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
formants.py
+3 -2
View File
@@ -9,17 +9,18 @@ def predict_from_times(wav_filename, preds_filename, begin, end, csv_export=True
tmp_features_filename = "temp/" + next(tempfile._get_candidate_names()) + ".txt" tmp_features_filename = "temp/" + next(tempfile._get_candidate_names()) + ".txt"
print("Input Array Path: " + tmp_features_filename) print("Input Array Path: " + tmp_features_filename)
predictions = None
if begin > 0.0 or end > 0.0: if begin > 0.0 or end > 0.0:
print(wav_filename + " interval " + str(begin) + "-" + str(end) + ":") print(wav_filename + " interval " + str(begin) + "-" + str(end) + ":")
features.create_features(wav_filename, tmp_features_filename, begin, end) features.create_features(wav_filename, tmp_features_filename, begin, end)
load_estimation_model(tmp_features_filename, preds_filename, begin, end, csv_export=csv_export) predictions = load_estimation_model(tmp_features_filename, preds_filename, begin, end, csv_export=csv_export)
#easy_call("luajit load_estimation_model.lua " + tmp_features_filename + ' ' + preds_filename) #easy_call("luajit load_estimation_model.lua " + tmp_features_filename + ' ' + preds_filename)
else: else:
features.create_features(wav_filename, tmp_features_filename) features.create_features(wav_filename, tmp_features_filename)
easy_call("luajit load_tracking_model.lua " + tmp_features_filename + ' ' + preds_filename) easy_call("luajit load_tracking_model.lua " + tmp_features_filename + ' ' + preds_filename)
delete_temp_files() delete_temp_files()
return predictions
def predict_from_textgrid(wav_filename, preds_filename, textgrid_filename, textgrid_tier): def predict_from_textgrid(wav_filename, preds_filename, textgrid_filename, textgrid_tier):
print(wav_filename) print(wav_filename)