update

examples/wenet/toolbox_infer.py

@@ -38,31 +38,6 @@ def get_args():
     return args
 
 
-def load_sherpa_offline_recognizer(nn_model_file: str,
-                                   tokens_file: str,
-                                   sample_rate: int = 16000,
-                                   num_active_paths: int = 2,
-                                   decoding_method: str = "greedy_search",
-                                   num_mel_bins: int = 80,
-                                   frame_dither: int = 0,
-                                   ):
-    feat_config = sherpa.FeatureConfig(normalize_samples=False)
-    feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
-    feat_config.fbank_opts.mel_opts.num_bins = 80
-    feat_config.fbank_opts.frame_opts.dither = 0
-
-    config = sherpa.OfflineRecognizerConfig(
-        nn_model=nn_model_file,
-        tokens=tokens_file,
-        use_gpu=False,
-        feat_config=feat_config,
-        decoding_method="greedy_search",
-        num_active_paths=2,
-    )
-    recognizer = sherpa.OfflineRecognizer(config)
-    return recognizer
-
-
 def main():
     args = get_args()
 

toolbox/k2_sherpa/models.py

@@ -56,7 +56,7 @@ def download_model(repo_id: str,
     return nn_model_file, tokens_file
 
 
-def load_sherpa_offline_recognizer(nn_model_file: str,
+def load_sherpa_offline_recognizer22(nn_model_file: str,
                                    tokens_file: str,
                                    sample_rate: int = 16000,
                                    num_active_paths: int = 2,
@@ -66,8 +66,8 @@ def load_sherpa_offline_recognizer(nn_model_file: str,
                                    ):
     feat_config = sherpa.FeatureConfig(normalize_samples=False)
     feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
-    feat_config.fbank_opts.mel_opts.num_bins = 80
-    feat_config.fbank_opts.frame_opts.dither = 0
+    feat_config.fbank_opts.mel_opts.num_bins = num_mel_bins
+    feat_config.fbank_opts.frame_opts.dither = frame_dither
 
     config = sherpa.OfflineRecognizerConfig(
         nn_model=nn_model_file,
@@ -81,7 +81,7 @@ def load_sherpa_offline_recognizer(nn_model_file: str,
     return recognizer
 
 
-def load_sherpa_offline_recognizer22(nn_model_file: str,
+def load_sherpa_offline_recognizer(nn_model_file: str,
                                    tokens_file: str,
                                    sample_rate: int = 16000,
                                    num_active_paths: int = 2,
@@ -89,7 +89,7 @@ def load_sherpa_offline_recognizer22(nn_model_file: str,
                                    num_mel_bins: int = 80,
                                    frame_dither: int = 0,
                                    ):
-    feat_config = sherpa.FeatureConfig()
+    feat_config = sherpa.FeatureConfig(normalize_samples=False)
     feat_config.fbank_opts.frame_opts.samp_freq = sample_rate
     feat_config.fbank_opts.mel_opts.num_bins = num_mel_bins
     feat_config.fbank_opts.frame_opts.dither = frame_dither