model

deploy_model.py

@@ -1,3 +1,11 @@
+'''
+Author: wuxulong19950206 [email protected]
+Date: 2024-03-11 19:46:42
+LastEditors: wuxulong19950206 [email protected]
+LastEditTime: 2024-03-11 19:48:42
+FilePath: \apeech_rec\deploy_model.py
+Description: 这是默认设置,请设置`customMade`, 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
+'''
 import os
 
 from speech_model import ModelSpeech
@@ -22,7 +30,7 @@ def get_pretrained_model():
     feat = Spectrogram()
     ms = ModelSpeech(sm251bn, feat, max_label_length=64)
 
-    ms.load_model('save_models/SpeechModel251bn/' + sm251bn.get_model_name() + '.model.h5')
+    ms.load_model('models\SpeechModel251bn_epoch18.model.h5')
     return ms
 def decode(model,filename):
 
@@ -35,6 +43,6 @@ def not_use():
     str_pinyin = res
     res = ml.pinyin_to_text(str_pinyin)
     print('语音识别最终结果：\n', res)
-chinese_models = {"chinese":'save_models/SpeechModel251bn/' + sm251bn.get_model_name() + '.model.h5'}
+chinese_models = {"chinese":'models\SpeechModel251bn_epoch18.model.h5'}
 language_to_models = {
     "Chinese": list(chinese_models.keys())}

models/SpeechModel251bn_epoch18.model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4f8984e420632d94b0e5bb507d75ddef6b76ded3ef0f9076188e541023255c7c
+size 6746064