tensorflow超薄修改resnet模型

Question

我想修改resnet_v2.resnet_v2_50模型，以便将一个数字连接到pool5层。导入网络后，我可以在end_points变量中看到这些层。

    with slim.arg_scope(resnet_v2.resnet_arg_scope()):
        net, end_points = resnet_v2.resnet_v2_50(self.imageIn, num_classes = numClasses)

所以我可以访问不同的层

curr_conv1 = end_points['resnet_v2_50/conv1']
curr_pred = end_points['resnet_v2_50/predictions']
curr_block4 = end_points['mainQN/resnet_v2_50/block4/unit_3/bottleneck_v2']

但是我不能访问网络中池化层之后的最后一部分。

curr_pool5 = end_points['resnet_v2_50/pool5']

但是我可以在Tensorboard和resnet_v2_50的代码中看到有某种pool5层。我如何才能访问它，这样我才能修改它并将一个数字连接到它？

Answer 1

end_points['global_pool']

就是你想要的。这紧跟在代码中称为pool5的全局池层之后。

https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py#L214