| | |
|---|
| | | from funasr.utils import asr_utils, wav_utils, postprocess_utils |
|---|
| | | import pdb |
|---|
| | | |
|---|
| | | header_colors = '\033[95m' |
|---|
| | | end_colors = '\033[0m' |
|---|
| | | |
|---|
| | | global_asr_language: str = 'zh-cn' |
|---|
| | | global_sample_rate: Union[int, Dict[Any, int]] = { |
|---|
| | |
|---|
| | | |
|---|
| | | Examples: |
|---|
| | | >>> import soundfile |
|---|
| | | >>> speech2text = Speech2Text("asr_config.yml", "asr.pth") |
|---|
| | | >>> speech2text = Speech2Text("asr_config.yml", "asr.pb") |
|---|
| | | >>> audio, rate = soundfile.read("speech.wav") |
|---|
| | | >>> speech2text(audio) |
|---|
| | | [(text, token, token_int, hypothesis object), ...] |
|---|
| | |
|---|
| | | # Input as audio signal |
|---|
| | | if isinstance(speech, np.ndarray): |
|---|
| | | speech = torch.tensor(speech) |
|---|
| | | |
|---|
| | | |
|---|
| | | if(speech.dim()==3): |
|---|
| | | speech = torch.squeeze(speech, 2) |
|---|
| | | #speech = speech.unsqueeze(0).to(getattr(torch, self.dtype)) |
|---|
| | | speech = speech.to(getattr(torch, self.dtype)) |
|---|
| | | # lenghts: (1,) |
|---|
| | |
|---|
| | | ngram_weight: float = 0.9, |
|---|
| | | nbest: int = 1, |
|---|
| | | num_workers: int = 1, |
|---|
| | | param_dict: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | |
|---|
| | | def _forward(data_path_and_name_and_type, |
|---|
| | | raw_inputs: Union[np.ndarray, torch.Tensor] = None, |
|---|
| | | output_dir_v2: Optional[str] = None, |
|---|
| | | fs: dict = None, |
|---|
| | | param_dict: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | # 3. Build data-iterator |
|---|
| | | if data_path_and_name_and_type is None and raw_inputs is not None: |
|---|
| | |
|---|
| | | data_path_and_name_and_type, |
|---|
| | | dtype=dtype, |
|---|
| | | batch_size=batch_size, |
|---|
| | | fs=fs, |
|---|
| | | mc=True, |
|---|
| | | key_file=key_file, |
|---|
| | | num_workers=num_workers, |
|---|
| | | preprocess_fn=ASRTask.build_preprocess_fn(speech2text.asr_train_args, False), |
|---|
| | |
|---|
| | | return asr_result_list |
|---|
| | | |
|---|
| | | return _forward |
|---|
| | | |
|---|
| | | def set_parameters(language: str = None, |
|---|
| | | sample_rate: Union[int, Dict[Any, int]] = None): |
|---|
| | | if language is not None: |
|---|
| | | global global_asr_language |
|---|
| | | global_asr_language = language |
|---|
| | | if sample_rate is not None: |
|---|
| | | global global_sample_rate |
|---|
| | | global_sample_rate = sample_rate |
|---|
| | | |
|---|
| | | |
|---|
| | | def get_parser(): |
|---|
| | | parser = config_argparse.ArgumentParser( |
|---|
