def merge_frames(df_outputs, frame_list, filename):
speaker_list = df_outputs.columns.tolist()
annotation = Annotation()
for speaker in speaker_list:
seg_start = 0
seg_end = 0
for i, label in enumerate(df_outputs[speaker]):
if (label == 1) and (seg_start == 0):
seg_start = float(frame_list[i][0])
elif (label == 0) and (seg_start > 0):
seg_end = float(frame_list[i][1])
annotation[Segment(start=seg_start, end=seg_end)] = speaker
seg_start = 0
else:
seg_end = float(frame_list[i][1])
with open(
'/home/lucvanwyk/MetricEmbeddingNet/rttm_out/' + filename +
'.rttm', 'w') as f:
annotation.write_rttm(f)
return annotation
