[Fix] empty npy folder

9 months ago · cb05ca2979
parent 2d7cf7f0e6
commit cb05ca2979
1 changed files with 4 additions and 1 deletions
--- a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py
+++ b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py
@ -39,7 +39,10 @@ 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 = [
+            f for f in self.root.glob("*")
+            if f.is_dir() and list(f.glob("*.npy"))
+        ]

        speaker_utterances = {
            speaker_dir: list(speaker_dir.glob("*.npy"))