Method __init__

ppocr/modeling/transforms/tsrn.py:170–177

(self, in_channels, up_scale)

class UpsampleBLock(nn.Layer):
    def __init__(self, in_channels, up_scale):
        super(UpsampleBLock, self).__init__()
        self.conv = nn.Conv2D(
            in_channels, in_channels * up_scale**2, kernel_size=3, padding=1
        )

        self.pixel_shuffle = nn.PixelShuffle(up_scale)
        self.prelu = mish()

    def forward(self, x):
        x = self.conv(x)