mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
25 lines
766 B
Python
25 lines
766 B
Python
from funasr.bin.diar_inference_launch import inference_launch
|
|
import sys
|
|
|
|
|
|
def main():
|
|
diar_config_path = sys.argv[1] if len(sys.argv) > 1 else "sond_fbank.yaml"
|
|
diar_model_path = sys.argv[2] if len(sys.argv) > 2 else "sond.pb"
|
|
output_dir = sys.argv[3] if len(sys.argv) > 3 else "./outputs"
|
|
data_path_and_name_and_type = [
|
|
("data/test_rmsil/feats.scp", "speech", "kaldi_ark"),
|
|
("data/test_rmsil/test_rmsil_tdnn6_xvec.scp", "profile", "kaldi_ark"),
|
|
]
|
|
pipeline = inference_launch(
|
|
mode="sond",
|
|
diar_train_config=diar_config_path,
|
|
diar_model_file=diar_model_path,
|
|
output_dir=output_dir,
|
|
num_workers=1
|
|
)
|
|
pipeline(data_path_and_name_and_type)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|