如何使用 dnnlib.tflib.network.Network 获得中间层 output

Question

我正在尝试StyleGAN2 NviLab，我想边生成边看特征图，但是我怎样才能得到中间层output？ model 由 dnnlib.tflib.network.Network 加载，没有关于此库的文档。

Answer 1

我有完全相同的问题，使用完全相同的代码库。 这就是我获得中间输出的方式（TF 1.14.0）：

步骤1：

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

network.print_layers()

第2步：

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

layers = network.list_layers()

第 3 步：

使用您需要的 output 的层的名称获取对代表层 output 的张量的引用：

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})