Ejemplo n.º 1
0
 def __init__(self, keys: KeysCollection, channel_dim: int = 0) -> None:
     """
     Args:
         keys: keys of the corresponding items to be transformed.
             See also: :py:class:`monai.transforms.compose.MapTransform`
         channel_dim: which dimension of input image is the channel, default is the first dimension.
     """
     super().__init__(keys)
     self.converter = AsChannelLast(channel_dim=channel_dim)
Ejemplo n.º 2
0
 def __init__(self, keys, channel_dim=0):
     """
     Args:
         keys (hashable items): keys of the corresponding items to be transformed.
             See also: :py:class:`monai.transforms.compose.MapTransform`
         channel_dim (int): which dimension of input image is the channel, default is the first dimension.
     """
     super().__init__(keys)
     self.converter = AsChannelLast(channel_dim=channel_dim)