funasr/models/campplus/utils.py
@@ -177,6 +177,9 @@ return res def smooth(res, mindur=1): # if only one segment, return directly if len(res) < 2: return res # short segments are assigned to nearest speakers. for i in range(len(res)): res[i][0] = round(res[i][0], 2) @@ -212,7 +215,7 @@ if overlap > max_overlap: max_overlap = overlap sentence_spk = spk d['spk'] = sentence_spk d['spk'] = int(sentence_spk) sd_sentence_list.append(d) return sd_sentence_list
