From 13e3fcec3016c86074f9902d3afae3f3bc1990ee Mon Sep 17 00:00:00 2001 From: xmyqsh Date: Mon, 17 Apr 2017 19:07:42 +0800 Subject: [PATCH] fix C%d to be the last residual block of each stage --- libs/nets/pyramid_network.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/nets/pyramid_network.py b/libs/nets/pyramid_network.py index 3dcc246..60fc6a9 100644 --- a/libs/nets/pyramid_network.py +++ b/libs/nets/pyramid_network.py @@ -18,10 +18,10 @@ # mapping each stage to its' tensor features _networks_map = { - 'resnet50': {'C1':'resnet_v1_50/conv1/Relu:0', - 'C2':'resnet_v1_50/block1/unit_2/bottleneck_v1', - 'C3':'resnet_v1_50/block2/unit_3/bottleneck_v1', - 'C4':'resnet_v1_50/block3/unit_5/bottleneck_v1', + 'resnet50': {#'C1':'resnet_v1_50/conv1', + 'C2':'resnet_v1_50/block1/unit_3/bottleneck_v1', + 'C3':'resnet_v1_50/block2/unit_4/bottleneck_v1', + 'C4':'resnet_v1_50/block3/unit_6/bottleneck_v1', 'C5':'resnet_v1_50/block4/unit_3/bottleneck_v1', }, 'resnet101': {'C1': '', 'C2': '',