未验证 提交 882dd964 编写于 作者: H haoyuying 提交者: GitHub

Fix the prediction problem of semantic segmentation module.

上级 8c1a18eb
......@@ -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)
......
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册