如何在tensorflow中从会话加载/卸载图形

时间:2018-03-12 19:04:43

标签: tensorflow

我正在研究我的一个项目,我必须将整个字母表的美国手语图像以及数字分类。

我从Tensorflow重新训练了Inception v3模型的最后一层。由于ASL中的某些数字符号与字母符号相同,因此我决定训练两种不同的模型。因此,我有两个日志文件夹:number_logsletters_logs。在这些文件夹中,您可以找到图形定义以及与类对应的标签(a,b,...和0,1,...)

我希望能够打开两个会话,或者找到一种方法来加载和卸载其中一个图表来对字母或数字进行分类。当前代码如下所示:

import tensorflow as tf

# Example with number classification
with tf.Session() as sess:
    self.label_lines = [lines.rstrip() for line in \
        tf.gfile.GFile('numbers_logs/trained_labels.txt')]

    with tf.gfile.FastGFile('numbers_logs/trained_graph.pb', 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        _ = tf.import_graph_def(self.graph_def, name='')
    softmax_tensor = sess.graph.get_tensor_by_name('final_result:0')

    image_data = cv2.imencode('.jpg', image)[1].tostring()
    predictions = sess.run(softmax_tensor, {'DecodeJpeg/contents:0': image_data})
    top_k = predictions[0].argsort()[-len(predictions[0]):][::-1]
    classified_symbol = label_lines[top_k[0]]

1 个答案:

答案 0 :(得分:0)

经过一番徘徊后,我发现运行两个不同图表的会话都能正常工作。我似乎无法通过不同的图表关闭会话并重新打开它。

SELECT j.Emailname, j.JobID
FROM   _Job j
WHERE  j.Emailname LIKE 'offer%'
AND EXIST(
    SELECT 1
    FROM   _Open o
    WHERE  o.JobID = j.JobID
)