Atomie CHEN
2024-03-25 ff83116369ba6afe56430cb0b9d977fbe7f4bced
fix: resolve IndexError when using spk model and the audio contains only 1 segment (#1535)

1个文件已修改
3 ■■■■■ 已修改文件
funasr/models/campplus/utils.py 3 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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)