From 7d5ffc20e0820ee769306cb76b6d234f62e92464 Mon Sep 17 00:00:00 2001 From: pengmian Date: Thu, 5 Sep 2019 15:47:39 +0800 Subject: [PATCH] modify resnet some layers name --- pdseg/models/backbone/resnet.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/pdseg/models/backbone/resnet.py b/pdseg/models/backbone/resnet.py index 6d6cc196..8a509272 100644 --- a/pdseg/models/backbone/resnet.py +++ b/pdseg/models/backbone/resnet.py @@ -133,7 +133,13 @@ class ResNet(): if layers >= 50: for block in range(len(depth)): for i in range(depth[block]): - conv_name = "conv" + str(block + 2) + '_' + str(1 + i) + if layers in [101, 152] and block == 2: + if i == 0: + conv_name = "res" + str(block + 2) + "a" + else: + conv_name = "res" + str(block + 2) + "b" + str(i) + else: + conv_name = "conv" + str(block + 2) + '_' + str(1 + i) dilation_rate = get_dilated_rate(dilation_dict, block) if self.stem == 'pspnet': @@ -172,7 +178,7 @@ class ResNet(): else: for block in range(len(depth)): for i in range(depth[block]): - conv_name = "conv" + str(block + 2) + chr(97 + i) + conv_name = "res" + str(block + 2) + chr(97 + i) conv = self.basic_block( input=conv, num_filters=num_filters[block], -- GitLab