Class PSPModule

competitor_blocks.py:7–35

Reference: Zhao, Hengshuang, et al. *"Pyramid scene parsing network."*

from blocks import ConvBnAct,SEModule

class PSPModule(nn.Module):
    """
    Reference:
        Zhao, Hengshuang, et al. *"Pyramid scene parsing network."*
    """
    def __init__(self, in_channels, out_channels=128, sizes=(1, 2, 3, 6)):
        super(PSPModule, self).__init__()

        convs = []
        for size in sizes:
            convs.append(
                nn.Sequential(
                    nn.AdaptiveAvgPool2d(output_size=(size, size)),
                    ConvBnAct(in_channels,out_channels,apply_act=False)
                )
            )
        self.stages=nn.ModuleList(convs)
        self.bottleneck=ConvBnAct(in_channels+len(sizes)*out_channels,out_channels)
        self.dropout=nn.Dropout2d(0.1)

    def forward(self, x):
        y=[x]
        for stage in self.stages:
            z=stage(x)
            z=F.interpolate(z,size=x.shape[-2:],align_corners=False,mode="bilinear")
            y.append(z)
        x=torch.cat(y,1)
        x = self.bottleneck(x)
        return x
class AlignedModule(nn.Module):
    #SFNet-DFNet
    def __init__(self, inplane, outplane):