Upload primary_doverlap.py

primary_doverlap.py

@@ -0,0 +1,25 @@
+import os
+import subprocess
+import dover_lap
+
+
+
+rttm_conf_dir = '' # path to the seg_CONF model output (RTTM file)
+
+rttm_fa_dir = '' # path to the seg_FA model output (RTTM file)
+
+rttm_miss_dir = '' # path to the seg_MISS model output (RTTM file)
+
+rttm_output_dir = '' # path to DOVER-Lap output (RTTM file)
+
+if not os.path.exists(rttm_output_dir):
+    os.makedirs(rttm_output_dir)
+
+
+bashCommand_doverlap = "dover-lap" + " " + rttm_output_dir + " " + rttm_conf_dir + " " + rttm_fa_dir + " " + rttm_miss_dir
+
+weights = " " + "--weight-type custom --custom-weight '[0.37,0.26,0.37]'" # custom weights based on the RTVE database
+
+bashCommand = bashCommand_doverlap + weights
+
+subprocess.run(bashCommand, shell=True, capture_output=True)