diff --git a/modules/image/semantic_segmentation/deeplabv3p_resnet50_voc/module.py b/modules/image/semantic_segmentation/deeplabv3p_resnet50_voc/module.py index 38cfd42978279008bb860071301482f6ace145d4..10ab03cae0b3759fc366b45667dd42baf8d8b3c9 100644 --- a/modules/image/semantic_segmentation/deeplabv3p_resnet50_voc/module.py +++ b/modules/image/semantic_segmentation/deeplabv3p_resnet50_voc/module.py @@ -76,7 +76,7 @@ class DeepLabV3PResnet50(nn.Layer): backbone_channels, aspp_ratios, aspp_out_channels, align_corners) self.align_corners = align_corners - self.transforms = T.Compose([T.Padding(target_size=(512, 512)), T.Normalize()]) + self.transforms = T.Compose([T.Normalize()]) if pretrained is not None: model_dict = paddle.load(pretrained) diff --git a/modules/image/semantic_segmentation/ocrnet_hrnetw18_voc/module.py b/modules/image/semantic_segmentation/ocrnet_hrnetw18_voc/module.py index 1660d1c607db7778d4ea3b37f4dfcecd16344700..1973f97426bb4bc9dbb05b99f20ccb9f807e54e9 100644 --- a/modules/image/semantic_segmentation/ocrnet_hrnetw18_voc/module.py +++ b/modules/image/semantic_segmentation/ocrnet_hrnetw18_voc/module.py @@ -69,7 +69,7 @@ class OCRNetHRNetW18(nn.Layer): ocr_mid_channels=ocr_mid_channels, ocr_key_channels=ocr_key_channels) self.align_corners = align_corners - self.transforms = T.Compose([T.Padding(target_size=(512, 512)), T.Normalize()]) + self.transforms = T.Compose([T.Normalize()]) if pretrained is not None: model_dict = paddle.load(pretrained)