凌匀
2023-03-13 c81e3096b2a445fbcfb6bb450efbd3f7d57822dd
fix the output of vad_results is null
1个文件已修改
15 ■■■■ 已修改文件
funasr/bin/vad_inference_online.py 15 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/vad_inference_online.py
@@ -1,4 +1,4 @@
import argparse
mport argparse
import logging
import sys
import json
@@ -236,12 +236,13 @@
            # param_dict['in_cache'] = batch['in_cache']
            if results:
                for i, _ in enumerate(keys):
                    results[i] = json.dumps(results[i])
                    item = {'key': keys[i], 'value': results[i]}
                    vad_results.append(item)
                    if writer is not None:
                        results[i] = json.loads(results[i])
                        ibest_writer["text"][keys[i]] = "{}".format(results[i])
                    if results[i]:
                        results[i] = json.dumps(results[i])
                        item = {'key': keys[i], 'value': results[i]}
                        vad_results.append(item)
                        if writer is not None:
                            results[i] = json.loads(results[i])
                            ibest_writer["text"][keys[i]] = "{}".format(results[i])
        return vad_results