InHeadConv3D

class aixd.mlmodel.architecture.heads.InHeadConv3D(*args: Any, **kwargs: Any)[source]

Convolutional Network for encoding spatially strucured (MRI- or video-like) data in 3 dimensions.

Parameters:

in_shape (int) – The shape of the input data.
latent_dims (List[int]) – A list of integers representing the dimensions of the latent space. The length of the sequence defines the number of layers.
activation (Union[nn.Module, str]) – The activation function to use.
attn_block_indices (List[int], optional, default=[]) – Sequence of indices where to insert an attention layer in the encoding blocks.

Inherited Methods