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()
|
