如何使用dnnlib.tflib.network.Network获取中间层输出

Question

我正在尝试StyleGAN2 NviLab，我想在生成的同时观察特征地图，但我如何才能获得中间层输出？这个模型是由dnnlib.tflib.network.Network加载的，没有关于这个库的文档。

Answer 1

我有完全相同的问题，使用完全相同的代码库。这就是我如何获得中间输出(TF 1.14.0)：

步骤1：

检查所有层的名称，以及它们的输出形状。这有助于识别您想要输出的图层的名称：

network.print_layers()

第2步：

获取网络中所有层的列表。该列表中的每个元素都是一个元组(layer name: string, layer output: tensor, layer variables)

layers = network.list_layers()

第3步：

获取表示层输出的张量的引用，使用您需要其输出的层的名称：

for layer in layers:
    print(layer[0], ' ', layer[1])
    if layer[0] == REQUIRED_LAYER_NAME:
        tensor = layer[1]

最后，运行图表：

required_output = sess.run(tensor, feed_dict={'INPUT_TENSOR_NAME': input_numpy_array})