tensorflow slim修改resnet模型

时间:2018-04-24 05:31:10

标签: python tensorflow tensorflow-slim

我想修改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层。我怎样才能访问它,所以我可以修改它并将数字连接到它?

1 个答案:

答案 0 :(得分:0)

end_points['global_pool']

是你想要的。紧接着代码中名为pool5的全局池层。

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