From f383666c400dcae1dc04f4081966255d27c8011a Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Mon, 18 Oct 2021 14:53:32 +0800 Subject: [PATCH] Add doc about how to check and use extracted alignments. --- egs/librispeech/ASR/conformer_ctc/README.md | 24 +++++++++++++++++++++ test/test_ali.py | 12 +++++------ 2 files changed, 30 insertions(+), 6 deletions(-) diff --git a/egs/librispeech/ASR/conformer_ctc/README.md b/egs/librispeech/ASR/conformer_ctc/README.md index 164c3e53e..cafc63bb6 100644 --- a/egs/librispeech/ASR/conformer_ctc/README.md +++ b/egs/librispeech/ASR/conformer_ctc/README.md @@ -51,3 +51,27 @@ in `conformer_ctc/train.py`. Search `./conformer_ctc/asr_datamodule.py` for `preserve_id`. **TODO:** Add doc about how to use the extracted alignment in the other pull-request. + +### Step 3: Check your extracted alignments + +There is a file `test_ali.py` in `icefall/test` that can be used to test your +alignments. It uses pre-computed alignments to modify a randomly generated +`nnet_output` and it checks that we can decode the correct transcripts +from the resulting `nnet_output`. + +You should get something like the following if you run that script: + +``` +$ ./test/test_ali.py +['THE GOOD NATURED AUDIENCE IN PITY TO FALLEN MAJESTY SHOWED FOR ONCE GREATER DEFERENCE TO THE KING THAN TO THE MINISTER AND SUNG THE PSALM WHICH THE FORMER HAD CALLED FOR', 'THE OLD SERVANT TOLD HIM QUIETLY AS THEY CREPT BACK TO DWELL THAT THIS PASSAGE THAT LED FROM THE HUT IN THE PLEASANCE TO SHERWOOD AND THAT GEOFFREY FOR THE TIME WAS HIDING WITH THE OUTLAWS IN THE FOREST', 'FOR A WHILE SHE LAY IN HER CHAIR IN HAPPY DREAMY PLEASURE AT SUN AND BIRD AND TREE', "BUT THE ESSENCE OF LUTHER'S LECTURES IS THERE"] +['THE GOOD NATURED AUDIENCE IN PITY TO FALLEN MAJESTY SHOWED FOR ONCE GREATER DEFERENCE TO THE KING THAN TO THE MINISTER AND SUNG THE PSALM WHICH THE FORMER HAD CALLED FOR', 'THE OLD SERVANT TOLD HIM QUIETLY AS THEY CREPT BACK TO GAMEWELL THAT THIS PASSAGE WAY LED FROM THE HUT IN THE PLEASANCE TO SHERWOOD AND THAT GEOFFREY FOR THE TIME WAS HIDING WITH THE OUTLAWS IN THE FOREST', 'FOR A WHILE SHE LAY IN HER CHAIR IN HAPPY DREAMY PLEASURE AT SUN AND BIRD AND TREE', "BUT THE ESSENCE OF LUTHER'S LECTURES IS THERE"] +``` + +### Step 4: Use your alignments in training + +Please refer to `conformer_mmi/train.py` for how usage. Some useful +functions are: + +- `load_alignments()`, it loads alignment saved by `conformer_ctc/ali.py` +- `convert_alignments_to_tensor()`, it converts alignments to PyTorch tensors +- `lookup_alignments()`, it returns the alignments of utterances by giving the cut ID of the utterances. diff --git a/test/test_ali.py b/test/test_ali.py index e8516e6dc..d8ada33e8 100755 --- a/test/test_ali.py +++ b/test/test_ali.py @@ -45,12 +45,12 @@ ICEFALL_DIR = Path(__file__).resolve().parent.parent egs_dir = ICEFALL_DIR / "egs/librispeech/ASR" lang_dir = egs_dir / "data/lang_bpe_500" # cut_json = egs_dir / "data/fbank/cuts_train-clean-100.json.gz" -cut_json = egs_dir / "data/fbank/cuts_train-clean-360.json.gz" +# cut_json = egs_dir / "data/fbank/cuts_train-clean-360.json.gz" # cut_json = egs_dir / "data/fbank/cuts_train-other-500.json.gz" -ali_filename = ICEFALL_DIR / "egs/librispeech/ASR/data/ali_500/train-960.pt" +# ali_filename = ICEFALL_DIR / "egs/librispeech/ASR/data/ali_500/train-960.pt" -# cut_json = egs_dir / "data/fbank/cuts_test-clean.json.gz" -# ali_filename = ICEFALL_DIR / "egs/librispeech/ASR/data/ali_500/test_clean.pt" +cut_json = egs_dir / "data/fbank/cuts_test-clean.json.gz" +ali_filename = ICEFALL_DIR / "egs/librispeech/ASR/data/ali_500/test_clean.pt" def data_exists(): @@ -62,7 +62,7 @@ def get_dataloader(): cuts_train = cuts_train.with_features_path_prefix(egs_dir) train_sampler = SingleCutSampler( cuts_train, - max_duration=200, + max_duration=40, shuffle=False, ) @@ -162,7 +162,7 @@ def test(): lattice = get_lattice( nnet_output=nnet_output, - HLG=HLG, + decoding_graph=HLG, supervision_segments=supervision_segments, search_beam=20, output_beam=8,