Тензорфлоу слим изменить модель 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