A python (3.6+) wrapper for Kaldi's data accessing.
- Kaldi's binary archives (*.ark)
- Kaldi's scripts (alignments & features, *.scp)
- Kaldi nnet3 data examples in binary (*.egs)
python setup.py install
or pip install kaldi-python-io
-
ArchiveReader && AlignArchiveReader
# allow only sequential index ark_reader = ArchiveReader("copy-feats ark:foo.ark ark:- |") for key, _ in ark_reader: print(key) ali_reader = AlignArchiveReader("gunzip -c foo.ali.gz |") for key, _ in ark_reader: print(key)
-
Nnet3EgsReader
# allow only sequential index egs_reader = Nnet3EgsReader("foo.egs") for key, _ in egs_reader: print(key)
-
ArchiveWriter
with ArchiveWriter("foo.ark", "foo.scp") as writer: for i in range(10): mat = np.random.rand(100, 20) writer.write(f"mat-{i}", mat)
-
ScriptReader && AlignScriptReader
# allow sequential/random index scp_reader = ScriptReader("shuf foo.scp | head -n 2") for key, mat in scp_reader: print(f"{key}: {mat.shape}") ali_reader = AlignScriptReader("foo.ali.scp") for key, ali in ali_reader: print(f"{key}: {ali.shape}")