相关疑难解决方法(0)

您好我想从tensorflow中微调VGG模型.我有两个问题.

如何从网络获得权重？trainable_variables为我返回空列表.

我从这里使用现有的模型:https://github.com/ry/tensorflow-vgg16.我发现了关于获取权重的帖子,但由于import_graph_def这对我不起作用.在TensorFlow训练的模型中获取一些权重的值

import tensorflow as tf
import PIL.Image
import numpy as np

with open("../vgg16.tfmodel", mode='rb') as f:
  fileContent = f.read()

graph_def = tf.GraphDef()
graph_def.ParseFromString(fileContent)

images = tf.placeholder("float", [None, 224, 224, 3])

tf.import_graph_def(graph_def, input_map={ "images": images })
print("graph loaded from disk")

graph = tf.get_default_graph()

cat = np.asarray(PIL.Image.open('../cat224.jpg'))
print(cat.shape)
init = tf.initialize_all_variables()

with tf.Session(graph=graph) as sess:
  print(tf.trainable_variables() )
  sess.run(init)