Exemplos de get_deconv_block em Python

Exemplo n.º 1

Arquivo: regunet.py Projeto: ETS-Research-Repositories/MONAI

    def build_up_sampling_block(self, in_channels: int, out_channels: int) -> nn.Module:
        if self._use_additive_upsampling:
            return AdditiveUpSampleBlock(
                spatial_dims=self.spatial_dims, in_channels=in_channels, out_channels=out_channels
            )

        return get_deconv_block(spatial_dims=self.spatial_dims, in_channels=in_channels, out_channels=out_channels)

Exemplo n.º 2

 def __init__(
     self,
     spatial_dims: int,
     in_channels: int,
     out_channels: int,
 ):
     super(AdditiveUpSampleBlock, self).__init__()
     self.deconv = get_deconv_block(spatial_dims=spatial_dims, in_channels=in_channels, out_channels=out_channels)

Exemplo n.º 3

 def build_up_sampling_block(
     self,
     in_channels: int,
     out_channels: int,
 ) -> nn.Module:
     return get_deconv_block(spatial_dims=self.spatial_dims,
                             in_channels=in_channels,
                             out_channels=out_channels)

Exemplo n.º 4

Arquivo: regunet.py Projeto: Nic-Ma/MONAI

 def __init__(self, spatial_dims: int, in_channels: int, out_channels: int):
     super().__init__()
     self.deconv = get_deconv_block(spatial_dims=spatial_dims,
                                    in_channels=in_channels,
                                    out_channels=out_channels)