我已经下载了一个实现VGG16 ConvNet的tensorflow GraphDef,我使用它来执行以下操作:
Pl['images'] = tf.placeholder(tf.float32,
[None, 448, 448, 3],
name="images") #batch x width x height x channels
with open("tensorflow-vgg16/vgg16.tfmodel", mode='rb') as f:
fileContent = f.read()
graph_def = tf.GraphDef()
graph_def.ParseFromString(fileContent)
tf.import_graph_def(graph_def, input_map={"images": Pl['images']})
Run Code Online (Sandbox Code Playgroud)
此外,我具有与的输出同质的图像特征"import/pool5/"。
我怎么能告诉我的图不想使用他的输入"images",而只是张量"import/pool5/"作为输入?
谢谢 !
编辑
好吧,我知道我还不太清楚。情况如下:
我正在尝试使用GraphDef格式的预训练VGG16来实现 ROI池的这种实现。所以这是我的工作:
首先,我加载模型:
tf.reset_default_graph()
with open("tensorflow-vgg16/vgg16.tfmodel",
mode='rb') as f:
fileContent = f.read()
graph_def = tf.GraphDef()
graph_def.ParseFromString(fileContent)
graph = tf.get_default_graph()
Run Code Online (Sandbox Code Playgroud)
然后,我创建我的占位符
images = tf.placeholder(tf.float32,
[None, 448, 448, 3],
name="images") #batch x width x …Run Code Online (Sandbox Code Playgroud)