预训练ANN的张量名称

时间:2019-03-05 10:35:46

标签: python tensorflow tensor

我正在使用https://colab.research.google.com/github/tensorflow/lucid/blob/master/notebooks/differentiable-parameterizations/xy2rgb.ipynb

中的CPPN代码进行功能可视化

我想找到一个可以用下面的代码替换“ mixed4a_pre_relu”的对象集。到目前为止,我已经了解到,“ mixed4a_pre_relu”是张量。当我尝试更改它的试错样式时,有时会很幸运,但有时会遇到这样的错误:

  

KeyError:“名称'import / mixed7a_pre_relu:0'表示不存在的张量。图形中不存在操作'import / mixed7a_pre_relu'。”

def render_feature(
    cppn_f = lambda: image_cppn(224),
    optimizer = tf.train.AdamOptimizer(0.005),
    objective = objectives.channel("mixed4a_pre_relu", 1)):
    vis = render.render_vis(model, objective, param_f=cppn_f, optimizer=optimizer, transforms=[], thresholds=[2**i for i in range(5,10)], verbose=False)
    show(vis)

render_feature()

我能找到的最佳解决方案是以下代码。

tf.reset_default_graph()

model = vision_models.InceptionV1()
model.load_graphdef()

for tensor in tf.get_default_graph().as_graph_def().node:
    print(tensor.name)

这根本不会给我起任何名字。当我在定义CPPN之后稍后尝试调用Tensor名称时,我只会得到CPPN的Tensor名称,而不是InceptionV1的Tensor名称。

如何找到预训练的CNN的列表目标/张量名称?

0 个答案:

没有答案