AudiogramDigitization / src /extract_ground_truth.py
weiren119's picture
Feat: app.py
34acdd0
raw
history blame contribute delete
765 Bytes
import json
import os
from digitizer.digitization import annotation_to_thresholds
if __name__ == "__main__":
for filename in os.listdir("../data/annotations_test"):
thresholds = annotation_to_thresholds(json.load(open(f"../data/annotations_test/{filename}")))
thresholds_list = []
for threshold in thresholds:
thresholds_list.append([threshold["ear"], threshold["conduction"], threshold["masking"], threshold["frequency"], threshold["threshold"]])
with open(f"../data/ground_truth/{filename.split('.')[0]}.csv", "w") as ofile:
ofile.write(f"ear,conduction,masking,frequency,threshold\n")
for t in thresholds_list:
ofile.write(f"{t[0]},{t[1]},{t[2]},{t[3]},{t[4]}\n")