From 2cdb2d654f2109ef4e648bae6f169143e267e5db Mon Sep 17 00:00:00 2001
From: zhuzizyf <42790740+zhuzizyf@users.noreply.github.com>
Date: 星期六, 11 三月 2023 14:33:14 +0800
Subject: [PATCH] Update dataset.py
---
egs/alimeeting/diarization/sond/unit_test.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/egs/alimeeting/diarization/sond/unit_test.py b/egs/alimeeting/diarization/sond/unit_test.py
index 10fc558..84a4247 100644
--- a/egs/alimeeting/diarization/sond/unit_test.py
+++ b/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,
--
Gitblit v1.9.1