我想修改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层。我怎样才能访问它,所以我可以修改它并将数字连接到它?
答案 0 :(得分:0)
end_points['global_pool']
是你想要的。紧接着代码中名为pool5的全局池层。
https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py#L214