如何获取SLIM vgg16网络中特定层的输出?

2024-06-16 13:45:22 发布

您现在位置:Python中文网/ 问答频道 /正文

我想在薄型vgg16网络中得到fc6层的输出,有什么好办法吗?在

其实我已经想出了一个可能的解决方案,请帮我确认一下它的正确性。在

fc6层的输出实际上是Relu操作的结果,即激活函数。通过执行tf.get_default_graph().get_operations(),我发现它的名字是vgg_16/fc6/Relu。所以,也许tf.get_default_graph().get_tensor_by_name('vgg_16/fc6/Relu:0')的结果就是我想要的?在


Tags: 函数网络defaultgettf解决方案graphoperations
1条回答
网友
1楼 · 发布于 2024-06-16 13:45:22

{TF}你能从这一层的图像中提取出什么样的图像

    from tensorflow.contrib.slim import nets

    last_layer_logits, end_points = nets.vgg.vgg_16(img_batch, num_classes=num_classes)

    # examples
    pool4_features = end_points['vgg_16/pool4']
    fc8_features = end_points['vgg_16/fc8']

如果您在实际vgg_16型号的基础上使用另一个argscope,则应将其添加到vgg_16 argscope之前:

^{pr2}$

希望这有帮助!在

相关问题 更多 >