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

[-] Remove CSV output

Browse Source
This commit is contained in:
Azalea (on HyDEV-Daisy)
2022-04-07 15:50:30 -04:00
parent 1bfd4c4346
commit 689e7d5e83
2 changed files with 8 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
load_estimation_model.py
+4 -12
View File
@@ -57,17 +57,9 @@ def load_estimation_model(inputfilename, outputfilename, begin, end, csv_export=
model.load_state_dict(torch.load("em.pth")) model.load_state_dict(torch.load("em.pth"))
my_prediction = model.forward(data) my_prediction = model.forward(data)
prediction_dict = {} prediction_dict = {"f1": 1000 * float(my_prediction[0][0]),
prediction_dict["F1"] = 1000 * float(my_prediction[0][0]) "f2": 1000 * float(my_prediction[0][1]),
prediction_dict["F2"] = 1000 * float(my_prediction[0][1]) "f3": 1000 * float(my_prediction[0][2]),
prediction_dict["F3"] = 1000 * float(my_prediction[0][2]) "f4": 1000 * float(my_prediction[0][3])}
prediction_dict["F4"] = 1000 * float(my_prediction[0][3])
if csv_export:
with open(outputfilename, "w") as wf:
wf.write("NAME,begin,end,F1,F2,F3,F4\n")
wf.write(name + "," + str(begin) + "," + str(end) + "," + \
str(prediction_dict["F1"]) + "," + str(prediction_dict["F2"]) + "," + \
str(prediction_dict["F3"]) + "," + str(prediction_dict["F4"]) + "\n")
return prediction_dict return prediction_dict
run.py
+4
View File
@@ -0,0 +1,4 @@
from formants import predict_from_times
if __name__ == '__main__':
predict_from_times('data/Example.wav', 'data/ExamplePredictions.csv', 0, -1)