Class autoencoder_vgg2

script/feature/model.py:69–111

add skip connections

        return encode, decode

class autoencoder_vgg2(nn.Module): # psnr 25.41
    ''' add skip connections '''
    def __init__(self):
        super(autoencoder_vgg2, self).__init__()
        conv1 = nn.Sequential(
            nn.Conv2d(3, 16, 3, stride=2, padding=1),
            nn.ReLU(inplace=True),
        )
        conv2 = nn.Sequential(
            nn.Conv2d(16, 32, 3, stride=2, padding=1),
            nn.ReLU(inplace=True),
        )
        conv3 = nn.Sequential(
            nn.Conv2d(32, 64, 7)
        )
        self.encoder = nn.Sequential(
            conv1, conv2, conv3
        )

        deconv1 = nn.Sequential(
            nn.ConvTranspose2d(64, 32, 7),
            nn.ReLU(inplace=True),
        )
        deconv2 = nn.Sequential(
            nn.ConvTranspose2d(32, 16, 3, stride=2, padding=1, output_padding=1),
            nn.ReLU(inplace=True),
        )
        deconv3 = nn.Sequential(
            nn.ConvTranspose2d(16, 3, 3, stride=2, padding=1, output_padding=1),
            nn.Tanh()
        )
        self.decoder = nn.Sequential(deconv1, deconv2, deconv3)
    def forward(self, x):
        feat1 = self.encoder[0](x)
        feat2 = self.encoder[1](feat1)
        x = self.encoder[2](feat2)

        x = self.decoder[0](x)
        x = x + feat2
        x = self.decoder[1](x)
        x = x + feat1
        x = self.decoder[2](x)
        return None, x

class autoencoder_vgg3(nn.Module): # psnr: 37.77
    ''' vgg encoder '''