diff --git a/pdseg/models/backbone/resnet.py b/pdseg/models/backbone/resnet.py index 6d6cc196f62186c3a8919b1b08e1df22d05d307a..8a509272df55d98e3d30fb9061e7c21788727df7 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],