Method __init__

x2ct_nerf/modules/image_encoder/resnet.py:256–261

(self, pretrained=True, in_channels=3, out_channels=1, latent_dim=128, input_img_size=320, **kwargs)

class ResNetAutoEncoder(nn.Module):
    def __init__(self, pretrained=True, in_channels=3, out_channels=1, latent_dim=128, input_img_size=320, **kwargs):
        super().__init__()
        self.in_channels = in_channels
        self.encoder = ResNetEncoder(pretrained=pretrained, in_channels=in_channels, latent_dim=latent_dim, input_img_size=input_img_size)
        self.decoder = Decoder(in_channels=latent_dim, out_channels=out_channels)
        self.transform = get_data_transform(in_channels=in_channels, input_img_size=input_img_size)

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