| | |
| | | import humanfriendly |
| | | import numpy as np |
| | | import torch |
| | | from torch_complex.tensor import ComplexTensor |
| | | from typeguard import check_argument_types |
| | | try: |
| | | from torch_complex.tensor import ComplexTensor |
| | | except: |
| | | print("Please install torch_complex firstly") |
| | | |
| | | from funasr.layers.log_mel import LogMel |
| | | from funasr.layers.stft import Stft |
| | | from funasr.models.frontend.abs_frontend import AbsFrontend |
| | | from funasr.modules.frontends.frontend import Frontend |
| | | from funasr.models.frontend.frontends_utils.frontend import Frontend |
| | | from funasr.utils.get_default_kwargs import get_default_kwargs |
| | | from funasr.modules.nets_utils import make_pad_mask |
| | | |
| | |
| | | apply_stft: bool = True, |
| | | use_channel: int = None, |
| | | ): |
| | | assert check_argument_types() |
| | | super().__init__() |
| | | if isinstance(fs, str): |
| | | fs = humanfriendly.parse_size(fs) |
| | |
| | | cmvn_file: str = None, |
| | | mc: bool = True |
| | | ): |
| | | assert check_argument_types() |
| | | super().__init__() |
| | | if isinstance(fs, str): |
| | | fs = humanfriendly.parse_size(fs) |