tensorflow从还原的模型获取变量名称

时间:2018-07-12 04:03:07

标签: python tensorflow deep-learning

我是tensorflow的新手。现在,我正在尝试还原mobielnetv2-SSD模型,修改模型结构并使用自己的数据对模型进行微调。我的问题是我无法从还原的模型中获取变量名。这是我的代码:

import tensorflow as tf 

with tf.Session() as sess:
    new_saver = tf.train.import_meta_graph('/home/dou/mobilenet-
    ssd/model_tf/v2.ckpt.meta')
    new_saver.restore(sess, '/home/dou/mobilenet-ssd/model_tf/v2.ckpt')
    graph = tf.get_default_graph()
    variables = graph.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)

但是当我检查变量列表时,它在调试控制台中始终为空列表。我的tensorflow版本是1.8。我在互联网上进行了大量搜索,但无法解决此问题。有人可以帮忙吗?谢谢。

0 个答案:

没有答案