alexey.zholtikov/mozc-devices
1
0
Fork 0
mirror of https://github.com/google/mozc-devices.git synced 2025-11-09 01:03:26 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update TF version

Browse Source
This commit is contained in:
Shuhei Iitsuka
2021-09-16 12:55:38 +09:00
parent 85c6365235
commit 1ecd083322
4 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mozc-nazoru/src/nazoru/predictor.py
View File

@@ -23,7 +23,7 @@ import numpy as np
def _load_graph(model_file):
graph = tf.Graph()
graph_def = tf.GraphDef()
graph_def = tf.compat.v1.GraphDef()
with open(model_file, "rb") as f:
graph_def.ParseFromString(f.read())
with graph.as_default():
@@ -44,7 +44,7 @@ class NazoruPredictor():
inputs = lib.keydowns2image(data, True, True, 16, 2)
inputs = np.expand_dims(inputs, axis=0)
with utils.Measure('sess.run'):
with tf.Session(graph=self._graph) as sess:
with tf.compat.v1.Session(graph=self._graph) as sess:
result = sess.run(self._output_operation.outputs[0],
{self._input_operation.outputs[0]: inputs})[0]
return result