小编Hed*_*iBY的帖子

Tensorflow:如何将自定义输入插入现有图形?

我已经下载了一个实现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)

graph subgraph tensorflow

5
推荐指数
1
解决办法
3704
查看次数

标签 统计

graph ×1

subgraph ×1

tensorflow ×1