Added TXT loading

sonogram_utility.py

@@ -100,6 +100,26 @@ def loadAudioRTTM(sampleRTTM):
 
     return speakerList, prediction
 
+def loadAudioTXT(sampleTXT):
+    # Read in prediction data
+    # Data in list form, for convenient plotting
+    speakerList = []
+    # Data in Annotation form, for convenient error rate calculation
+    prediction = Annotation(uri=sampleTXT)
+    with open(sampleTXT, "r") as txt:
+        for line in txt:
+            speakerResult = line.split('\t')
+            print(speakerResult)
+            if len(speakerResult) < 3:
+                continue
+            index = -1
+            start = float(speakerResult[0])
+            end = float(speakerResult[1])
+            duration = end - start
+            prediction[Segment(start,end)] = speakerResult[2]
+
+    return [], prediction
+
 def splitIntoTimeSegments(testFile,maxDurationInSeconds=60):
 
     waveform, sample_rate = torchaudio.load(testFile)