update

Dockerfile

@@ -11,6 +11,8 @@ RUN pip install --upgrade pip
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
 
+RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/
+
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 

README.md

@@ -42,6 +42,17 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/miniconda3/envs/ASR/lib/
 
 https://blog.csdn.net/qq_38225558/article/details/128641561
 ```text
-ImportError: /lib64/libm.so.6: version `GLIBC_2.27' not found
+ImportError: libk2_torch_api.so: cannot open shared object file: No such file or directory
+
+```
+
+查找 libk2_torch_api.so 共享库所在目录
+```text
+find / -name libk2_torch_api.so
 ```
 
+添加 LD_LIBRARY_PATH
+```text
+export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/
+
+```

examples/wenet/{test.py → downaload_model.py}

@@ -9,7 +9,6 @@ pwd = os.path.abspath(os.path.dirname(__file__))
 sys.path.append(os.path.join(pwd, "../../"))
 
 import huggingface_hub
-import sherpa
 
 from project_settings import project_path
 
@@ -61,23 +60,6 @@ def main():
         local_dir=model_dir.as_posix(),
     )
     print(tokens_filename)
-
-    feat_config = sherpa.FeatureConfig(normalize_samples=False)
-    feat_config.fbank_opts.frame_opts.samp_freq = 16000
-    feat_config.fbank_opts.mel_opts.num_bins = 80
-    feat_config.fbank_opts.frame_opts.dither = 0
-
-    config = sherpa.OfflineRecognizerConfig(
-        nn_model=model_filename,
-        tokens=tokens_filename,
-        use_gpu=False,
-        feat_config=feat_config,
-        decoding_method="greedy_search",
-        num_active_paths=2,
-    )
-
-    recognizer = sherpa.OfflineRecognizer(config)
-    print(recognizer)
     return
 
 

examples/wenet/infer.py

@@ -0,0 +1,54 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+import argparse
+import os
+from pathlib import Path
+import sys
+
+pwd = os.path.abspath(os.path.dirname(__file__))
+sys.path.append(os.path.join(pwd, "../../"))
+
+import sherpa
+
+from project_settings import project_path
+
+
+def get_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--model_dir",
+        default=(project_path / "pretrained_models/huggingface/csukuangfj/wenet-chinese-model").as_posix(),
+        type=str
+    )
+    args = parser.parse_args()
+    return args
+
+
+def main():
+    args = get_args()
+
+    model_dir = Path(args.model_dir)
+    model_filename = model_dir / "final.zip"
+    tokens_filename = model_dir / "units.txt"
+
+    feat_config = sherpa.FeatureConfig(normalize_samples=False)
+    feat_config.fbank_opts.frame_opts.samp_freq = 16000
+    feat_config.fbank_opts.mel_opts.num_bins = 80
+    feat_config.fbank_opts.frame_opts.dither = 0
+
+    config = sherpa.OfflineRecognizerConfig(
+        nn_model=model_filename.as_posix(),
+        tokens=tokens_filename.as_posix(),
+        use_gpu=False,
+        feat_config=feat_config,
+        decoding_method="greedy_search",
+        num_active_paths=2,
+    )
+
+    recognizer = sherpa.OfflineRecognizer(config)
+    print(recognizer)
+    return
+
+
+if __name__ == "__main__":
+    main()