我想找到一个可以用下面的代码替换“ 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的列表目标/张量名称?