add reset_default_graph.
修改获取tensor值的代码,添加重置TF 运行图的代码,解决了 预测函数多次运行越来越慢的问题。
This commit is contained in:
parent
b2ba6318cd
commit
14bd31212c
|
@ -12,6 +12,7 @@ from general_function.file_dict import *
|
||||||
from general_function.gen_func import *
|
from general_function.gen_func import *
|
||||||
|
|
||||||
# LSTM_CNN
|
# LSTM_CNN
|
||||||
|
import tensorflow as tf
|
||||||
import keras as kr
|
import keras as kr
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import random
|
import random
|
||||||
|
@ -322,19 +323,9 @@ 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())
|
||||||
|
tf.reset_default_graph()
|
||||||
r1 = K.get_value(r[0][0])
|
return r1[0]
|
||||||
#print('r1', r1)
|
|
||||||
|
|
||||||
|
|
||||||
#r2 = K.get_value(r[1])
|
|
||||||
#print(r2)
|
|
||||||
|
|
||||||
r1=r1[0]
|
|
||||||
|
|
||||||
return r1
|
|
||||||
pass
|
|
||||||
|
|
||||||
def RecognizeSpeech(self, wavsignal, fs):
|
def RecognizeSpeech(self, wavsignal, fs):
|
||||||
'''
|
'''
|
||||||
|
|
Loading…
Reference in New Issue