def forward(self, x, x_pre):
    main = (x + x_pre)
    main = self.main_conv1(main)
    main = F.interpolate(main, scale_factor=2, mode='bilinear', align_corners=True)
    ext = self.ext_conv1(x)
    ext = self.ext_conv2(ext)
    ext = self.ext_conv3(ext)
    ext = self.ext_regul(ext)
    out = self.out_prelu((main + ext))
    return out
def forward(self, x, x_pre):
    main = (x + x_pre)
    main = self.main_conv1(main)
    main = F.interpolate(main, scale_factor=2, mode='bilinear', align_corners=True)
    ext = self.ext_conv1(x)
    ext = self.ext_conv2(ext)
    ext = self.ext_conv3(ext)
    ext = self.ext_regul(ext)
    out = self.out_prelu((main + ext))
    return out