add reset_default_graph.

修改获取tensor值的代码,添加重置TF 运行图的代码,解决了 预测函数多次运行越来越慢的问题。
This commit is contained in:
张旭 2020-05-20 14:46:08 +08:00 committed by GitHub
parent b2ba6318cd
commit 14bd31212c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 13 deletions

View File

@ -12,6 +12,7 @@ from general_function.file_dict import *
from general_function.gen_func import *
# LSTM_CNN
import tensorflow as tf
import keras as kr
import numpy as np
import random
@ -322,19 +323,9 @@ class ModelSpeech(): # 语音模型类
r = K.ctc_decode(base_pred, in_len, greedy = True, beam_width=100, top_paths=1)
#print('r', r)
r1 = K.get_value(r[0][0])
#print('r1', r1)
#r2 = K.get_value(r[1])
#print(r2)
r1=r1[0]
return r1
pass
r1 = r[0][0].eval(session=tf.Session())
tf.reset_default_graph()
return r1[0]
def RecognizeSpeech(self, wavsignal, fs):
'''