游雁
2023-02-13 fcc9c89eaba9a4e36c54958aeedeec7ab3756cd7
egs/alimeeting/diarization/sond/unit_test.py
@@ -48,7 +48,7 @@
    diar_model_path = "sond.pth"
    output_dir = "./outputs"
    data_path_and_name_and_type = [
        ("data/unit_test/test_wav.scp", "speech", "kaldi_ark"),
        ("data/unit_test/test_wav.scp", "speech", "sound"),
        ("data/unit_test/test_profile.scp", "profile", "kaldi_ark"),
    ]
    pipeline = inference_launch(
@@ -68,13 +68,13 @@
    diar_config_path = "config.yaml"
    diar_model_path = "sond.pth"
    output_dir = "./outputs"
    raw_inputs = [
    raw_inputs = [[
        "data/unit_test/raw_inputs/record.wav",
        "data/unit_test/raw_inputs/spk1.wav",
        "data/unit_test/raw_inputs/spk2.wav",
        "data/unit_test/raw_inputs/spk3.wav",
        "data/unit_test/raw_inputs/spk4.wav"
    ]
    ]]
    pipeline = inference_launch(
        mode="sond_demo",
        diar_train_config=diar_config_path,