修复部分代码对tf2.0的兼容问题

This commit is contained in:
nl 2020-05-30 12:52:53 +08:00
parent 4dad3e73e1
commit 74d5e40dea
1 changed files with 2 additions and 2 deletions

View File

@ -323,8 +323,8 @@ class ModelSpeech(): # 语音模型类
r = K.ctc_decode(base_pred, in_len, greedy = True, beam_width=100, top_paths=1) r = K.ctc_decode(base_pred, in_len, greedy = True, beam_width=100, top_paths=1)
#print('r', r) #print('r', r)
r1 = r[0][0].eval(session=tf.Session()) r1 = r[0][0].eval(session=tf.compat.v1.Session())
tf.reset_default_graph() tf.compat.v1.reset_default_graph()
return r1[0] return r1[0]
def RecognizeSpeech(self, wavsignal, fs): def RecognizeSpeech(self, wavsignal, fs):