From 7d26f93d2c3b84bbad8a7f1b71b679f61520f05d Mon Sep 17 00:00:00 2001 From: megemini Date: Tue, 31 Dec 2024 14:32:22 +0800 Subject: [PATCH] =?UTF-8?q?[Hackathon=207th]=20=E4=BF=AE=E5=A4=8D=E4=B8=8D?= =?UTF-8?q?=E5=AD=98=E5=9C=A8=20`*.npy`=20=E6=96=87=E4=BB=B6=E7=9A=84?= =?UTF-8?q?=E7=A9=BA=E6=96=87=E4=BB=B6=E5=A4=B9=E5=AF=BC=E8=87=B4=E7=9A=84?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E9=81=8D=E5=8E=86=E9=94=99=E8=AF=AF=20(#3948?= =?UTF-8?q?)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * [Fix] empty npy folder * [Update] assert empyt folder --- .../vector/exps/ge2e/speaker_verification_dataset.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py index 87b7d59a3..ae2c9427f 100644 --- a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py +++ b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py @@ -39,7 +39,12 @@ class MultiSpeakerMelDataset(Dataset): def __init__(self, dataset_root: Path): self.root = Path(dataset_root).expanduser() - speaker_dirs = [f for f in self.root.glob("*") if f.is_dir()] + speaker_dirs = [] + for f in self.root.glob("*"): + if f.is_dir(): + assert list(f.glob( + "*.npy")), "This folder NOT includes any npy data file." + speaker_dirs.append(f) speaker_utterances = { speaker_dir: list(speaker_dir.glob("*.npy"))