33c2f89adf27cad1a5b0b0a9e542b989f1f9eed4,gluon/gluoncv2/models/pspnet.py,PSPNet,hybrid_forward,#PSPNet#Any#Any#,134

Before Change

    def hybrid_forward(self, F, x):
        x = self.backbone(x)
        x = self.pool(x)
        x = self.conv1(x)
        x = self.dropout(x)
        x = self.conv2(x)
        x = F.contrib.BilinearResize2D(x, height=self.in_size[0], width=self.in_size[1])
        return x


def get_pspnet(backbone,
               classes,

After Change

        x = outs[0]
        x = self.pool(x)
        x = self.final_block(x)
        if self.aux:
            y = outs[1]
            y = self.aux_block(y)
            return x, y
        else:
            return x


def get_pspnet(backbone,
               classes,
               model_name=None,
               pretrained=False,

Instances