Esempi in Python per get_conv_block

Esempio n. 1

 def build_bottom_block(self, in_channels: int, out_channels: int):
     kernel_size = self.encode_kernel_sizes[self.depth]
     return get_conv_block(
         spatial_dims=self.spatial_dims,
         in_channels=in_channels,
         out_channels=out_channels,
         kernel_size=kernel_size,
     )

Esempio n. 2

File: regunet.py Progetto: Nic-Ma/MONAI

 def build_conv_block(self, in_channels, out_channels, kernel_size):
     return nn.Sequential(
         get_conv_block(
             spatial_dims=self.spatial_dims,
             in_channels=in_channels,
             out_channels=out_channels,
             kernel_size=kernel_size,
         ),
         RegistrationResidualConvBlock(
             spatial_dims=self.spatial_dims,
             in_channels=out_channels,
             out_channels=out_channels,
             kernel_size=kernel_size,
         ),
     )

Esempio n. 3

File: regunet.py Progetto: Nic-Ma/MONAI

 def build_bottom_block(self, in_channels: int, out_channels: int):
     kernel_size = self.encode_kernel_sizes[self.depth]
     return nn.Sequential(
         get_conv_block(
             spatial_dims=self.spatial_dims,
             in_channels=in_channels,
             out_channels=out_channels,
             kernel_size=kernel_size,
         ),
         RegistrationResidualConvBlock(
             spatial_dims=self.spatial_dims,
             in_channels=out_channels,
             out_channels=out_channels,
             kernel_size=kernel_size,
         ),
     )