| | |
|---|
| | | import torch |
|---|
| | | import torchaudio.compliance.kaldi as kaldi |
|---|
| | | from torch.nn.utils.rnn import pad_sequence |
|---|
| | | from typeguard import check_argument_types |
|---|
| | | |
|---|
| | | import funasr.models.frontend.eend_ola_feature as eend_ola_feature |
|---|
| | | from funasr.models.frontend.abs_frontend import AbsFrontend |
|---|
| | |
|---|
| | | snip_edges: bool = True, |
|---|
| | | upsacle_samples: bool = True, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | super().__init__() |
|---|
| | | self.fs = fs |
|---|
| | | self.window = window |
|---|
| | |
|---|
| | | snip_edges: bool = True, |
|---|
| | | upsacle_samples: bool = True, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | super().__init__() |
|---|
| | | self.fs = fs |
|---|
| | | self.window = window |
|---|
| | |
|---|
| | | return feats_pad, feats_lens, lfr_splice_frame_idxs |
|---|
| | | |
|---|
| | | def forward( |
|---|
| | | self, input: torch.Tensor, input_lengths: torch.Tensor, is_final: bool = False, reset: bool = True |
|---|
| | | self, input: torch.Tensor, input_lengths: torch.Tensor, is_final: bool = False, reset: bool = False |
|---|
| | | ) -> Tuple[torch.Tensor, torch.Tensor]: |
|---|
| | | if reset: |
|---|
| | | self.cache_reset() |
|---|
| | |
|---|
| | | lfr_m: int = 1, |
|---|
| | | lfr_n: int = 1, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | super().__init__() |
|---|
| | | self.fs = fs |
|---|
| | | self.frame_length = frame_length |
|---|
