Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Index your code
hub
/
github.com/Project-MONAI/MONAI
/ types & classes
Types & classes
2,334 in github.com/Project-MONAI/MONAI
⨍
Functions
7,906
◇
Types & classes
2,334
↳
Endpoints
20
↓ 223 callers
Class
MetaTensor
Class that inherits from both `torch.Tensor` and `MetaObj`, adding support for metadata. Metadata is stored in the form of a dictionary. Nes
monai/data/meta_tensor.py:52
↓ 207 callers
Class
Compose
``Compose`` provides the ability to chain a series of callables together in a sequential manner. Each transform in the sequence must take a s
monai/transforms/compose.py:141
↓ 154 callers
Class
Convolution
Constructs a convolution with normalization, optional dropout, and optional activation layers:: -- (Conv|ConvTrans) -- (Norm -- Dropout
monai/networks/blocks/convolutions.py:25
↓ 95 callers
Class
LoadImaged
Dictionary-based wrapper of :py:class:`monai.transforms.LoadImage`, It can load both image data and metadata. When loading a list of files in
monai/transforms/io/dictionary.py:39
↓ 86 callers
Class
DataLoader
Provides an iterable over the given `dataset`. It inherits the PyTorch DataLoader and adds enhanced `collate_fn` and `worker_fn` by default.
monai/data/dataloader.py:26
↓ 67 callers
Class
LoadImage
Load image file or files from provided path based on reader. If reader is not specified, this class automatically chooses readers based o
monai/transforms/io/array.py:109
↓ 66 callers
Class
ConfigParser
The primary configuration parser. It traverses a structured config (in the form of nested Python dict or list), creates ``ConfigItem``, and a
monai/bundle/config_parser.py:237
↓ 65 callers
Class
EnsureChannelFirstd
Dictionary-based wrapper of :py:class:`monai.transforms.EnsureChannelFirst`.
monai/transforms/utility/dictionary.py:259
↓ 62 callers
Class
DiffusionModelUNet
Unet network with timestep embedding and attention mechanisms for conditioning based on Rombach et al. "High-Resolution Image Synthesis with
monai/networks/nets/diffusion_model_unet.py:1507
↓ 59 callers
Class
Dataset
A generic dataset with a length property and an optional callable data transform when fetching a data sample. If passing slicing indices,
monai/data/dataset.py:57
↓ 42 callers
Class
DDPMScheduler
Denoising diffusion probabilistic models (DDPMs) explores the connections between denoising score matching and Langevin dynamics sampling. Ba
monai/networks/schedulers/ddpm.py:67
↓ 42 callers
Class
SlidingWindowSplitter
Splits the input into patches with sliding window strategy and a possible overlap. It also allows offsetting the starting position and filter
monai/inferers/splitter.py:94
↓ 40 callers
Class
SurfaceDiceMetric
Computes the Normalized Surface Dice (NSD) for each batch sample and class of predicted segmentations `y_pred` and corresponding reference se
monai/metrics/surface_dice.py:26
↓ 39 callers
Class
AutoencoderKL
Autoencoder model with KL-regularized latent space based on Rombach et al. "High-Resolution Image Synthesis with Latent Diffusion Models" htt
monai/networks/nets/autoencoderkl.py:471
↓ 39 callers
Class
VQVAE
Vector-Quantised Variational Autoencoder (VQ-VAE) used in Morphology-preserving Autoregressive 3D Generative Modelling of the Brain by Tudosi
monai/networks/nets/vqvae.py:275
↓ 38 callers
Class
SPADEDiffusionModelUNet
UNet network with timestep embedding and attention mechanisms for conditioning, with added SPADE normalization for semantic conditioning (Par
monai/networks/nets/spade_diffusion_model_unet.py:673
↓ 37 callers
Class
FocalLoss
FocalLoss is an extension of BCEWithLogitsLoss that down-weights loss from high confidence correct predictions. Reimplementation of the
monai/losses/focal_loss.py:25
↓ 34 callers
Class
EnsureChannelFirst
Adjust or add the channel dimension of input data to ensure `channel_first` shape. This extracts the `original_channel_dim` info from provid
monai/transforms/utility/array.py:175
↓ 32 callers
Class
Spacingd
Dictionary-based wrapper of :py:class:`monai.transforms.Spacing`. This transform assumes the ``data`` dictionary has a key for the input
monai/transforms/spatial/dictionary.py:367
↓ 30 callers
Class
AffineTransform
monai/networks/layers/spatial_transforms.py:439
↓ 29 callers
Class
Range
A decorator and context manager for NVIDIA Tools Extension (NVTX) Range for profiling. When used as a decorator it encloses a specific method
monai/utils/nvtx.py:33
↓ 27 callers
Class
ToTensor
Converts the input image to a tensor without applying any other transformations. Input data can be PyTorch Tensor, numpy array, list, diction
monai/transforms/utility/array.py:383
↓ 26 callers
Class
DiceLoss
Compute average Dice loss between two tensors. It can support both multi-classes and multi-labels tasks. The data `input` (BNHW[D] where N is
monai/losses/dice.py:30
↓ 26 callers
Class
SPADEAutoencoderKL
Autoencoder model with KL-regularized latent space based on Rombach et al. "High-Resolution Image Synthesis with Latent Diffusion Models" htt
monai/networks/nets/spade_autoencoderkl.py:309
↓ 25 callers
Class
ScaleIntensityd
Dictionary-based wrapper of :py:class:`monai.transforms.ScaleIntensity`. Scale the intensity of input image to the given value range (minv, m
monai/transforms/intensity/dictionary.py:550
↓ 24 callers
Class
UNet
Enhanced version of UNet which has residual units implemented with the ResidualUnit class. The residual part uses a convolution to change the
monai/networks/nets/unet.py:28
↓ 23 callers
Class
SlidingWindowInferer
Sliding window method for model inference, with `sw_batch_size` windows for every model.forward(). Usage example can be found in the :py:
monai/inferers/inferer.py:446
↓ 23 callers
Class
SomeOf
``SomeOf`` samples a different sequence of transforms to apply each time it is called. It can be configured to sample a fixed or varying num
monai/transforms/compose.py:664
↓ 22 callers
Class
Flip
Reverses the order of elements along the given spatial axis. Preserves shape. See `torch.flip` documentation for additional details: http
monai/transforms/spatial/array.py:706
↓ 22 callers
Class
ImageFilter
Applies a convolution filter to the input image. Args: filter: A string specifying the filter, a custom filter as ``torc
monai/transforms/utility/array.py:1534
↓ 22 callers
Class
Invertd
Utility transform to invert the previously applied transforms. Taking the ``transform`` previously applied on ``orig_keys``, this ``Invertd`
monai/transforms/post/dictionary.py:772
↓ 22 callers
Class
OneOf
``OneOf`` provides the ability to randomly choose one transform out of a list of callables with pre-defined probabilities for each. Args
monai/transforms/compose.py:424
↓ 22 callers
Class
RFlowScheduler
A rectified flow scheduler for guiding the diffusion process in a generative model. Supports uniform and logit-normal sampling methods, time
monai/networks/schedulers/rectified_flow.py:78
↓ 22 callers
Class
SABlock
A self-attention block, based on: "Dosovitskiy et al., An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://ar
monai/networks/blocks/selfattention.py:26
↓ 21 callers
Class
ConfigWorkflow
Specification for the config-based bundle workflow. Standardized the `initialize`, `run`, `finalize` behavior in a config-based training, eva
monai/bundle/workflows.py:361
↓ 21 callers
Class
ControlNet
Control network for diffusion models based on Zhang and Agrawala "Adding Conditional Control to Text-to-Image Diffusion Models" (https://arxi
monai/networks/nets/controlnet.py:123
↓ 21 callers
Class
LossReduction
See also: - :py:class:`monai.losses.dice.DiceLoss` - :py:class:`monai.losses.dice.GeneralizedDiceLoss` - :py:class:`monai
monai/utils/enums.py:253
↓ 20 callers
Class
CacheDataset
Dataset with cache mechanism that can load data and cache deterministic transforms' result during training. By caching the results of non-ra
monai/data/dataset.py:733
↓ 20 callers
Class
ClipIntensityPercentiles
Apply clip based on the intensity distribution of input image. If `sharpness_factor` is provided, the intensity values will be soft clipped a
monai/transforms/intensity/array.py:1039
↓ 20 callers
Class
DiffusionModelUNetMaisi
U-Net network with timestep embedding and attention mechanisms for conditioning based on Rombach et al. "High-Resolution Image Synthesis with
monai/apps/generation/maisi/networks/diffusion_model_unet_maisi.py:53
↓ 20 callers
Class
GaussianFilter
monai/networks/layers/simplelayers.py:542
↓ 20 callers
Class
ITKReader
Load medical images based on ITK library. All the supported image formats can be found at: https://github.com/InsightSoftwareConsortium/I
monai/data/image_reader.py:169
↓ 19 callers
Class
AsDiscrete
Convert the input tensor/array into discrete values, possible operations are: - `argmax`. - threshold input value to binary va
monai/transforms/post/array.py:140
↓ 19 callers
Class
ClipIntensityPercentilesd
Dictionary-based wrapper of :py:class:`monai.transforms.ClipIntensityPercentiles`. Clip the intensity values of input image to a specific ran
monai/transforms/intensity/dictionary.py:906
↓ 19 callers
Class
DiffusionUNetResnetBlock
Residual block with timestep conditioning. Args: spatial_dims: The number of spatial dimensions. in_channels: number of inpu
monai/networks/nets/diffusion_model_unet.py:330
↓ 19 callers
Class
Lambdad
Dictionary-based wrapper of :py:class:`monai.transforms.Lambda`. For example: .. code-block:: python :emphasize-lines: 2
monai/transforms/utility/dictionary.py:1023
↓ 19 callers
Class
Rotate90
Rotate an array by 90 degrees in the plane specified by `axes`. See `torch.rot90` for additional details: https://pytorch.org/docs/stable
monai/transforms/spatial/array.py:1181
↓ 18 callers
Class
CrossAttentionBlock
A cross-attention block, based on: "Dosovitskiy et al., An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://a
monai/networks/blocks/crossattention.py:25
↓ 18 callers
Class
Dataset_
tests/data/test_zipdataset.py:22
↓ 18 callers
Class
Resize
Resize the input image to given spatial size (with scaling, not cropping/padding). Implemented using :py:class:`torch.nn.functional.interpola
monai/transforms/spatial/array.py:751
↓ 18 callers
Class
UltrasoundConfidenceMapTransform
Compute confidence map from an ultrasound image. This transform uses the method introduced by Karamalis et al. in https://doi.org/10.1016/j.media.
monai/transforms/intensity/array.py:2814
↓ 17 callers
Class
DiffusionInferer
DiffusionInferer takes a trained diffusion model and a scheduler and can be used to perform a signal forward pass for a training iteration, a
monai/inferers/inferer.py:850
↓ 17 callers
Class
Orientationd
Dictionary-based wrapper of :py:class:`monai.transforms.Orientation`. This transform assumes the channel-first input format. In the case
monai/transforms/spatial/dictionary.py:543
↓ 17 callers
Class
PersistentDataset
Persistent storage of pre-computed values to efficiently manage larger than memory dictionary format data, it can operate transforms for spec
monai/data/dataset.py:162
↓ 17 callers
Class
SpatialCrop
General purpose cropper to produce sub-volume region of interest (ROI). If a dimension of the expected ROI size is larger than the input imag
monai/transforms/croppad/array.py:437
↓ 17 callers
Class
SpatialCropd
Dictionary-based wrapper of :py:class:`monai.transforms.SpatialCrop`. General purpose cropper to produce sub-volume region of interest (ROI).
monai/transforms/croppad/dictionary.py:420
↓ 16 callers
Class
A
tests/transforms/compose/test_some_of.py:33
↓ 16 callers
Class
Activations
Activation operations, typically `Sigmoid` or `Softmax`. Args: sigmoid: whether to execute sigmoid function on model output before t
monai/transforms/post/array.py:69
↓ 16 callers
Class
SpatialAttentionBlock
Perform spatial self-attention on the input tensor. The input tensor is reshaped to B x (x_dim * y_dim [ * z_dim]) x C, where C is the number of
monai/networks/blocks/spatialattention.py:20
↓ 15 callers
Class
B
tests/transforms/compose/test_some_of.py:39
↓ 15 callers
Class
C
tests/transforms/compose/test_some_of.py:45
↓ 15 callers
Class
EnsureTyped
Dictionary-based wrapper of :py:class:`monai.transforms.EnsureType`. Ensure the input data to be a PyTorch Tensor or numpy array, support: `
monai/transforms/utility/dictionary.py:493
↓ 15 callers
Class
PatchEmbeddingBlock
A patch embedding block, based on: "Dosovitskiy et al., An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://a
monai/networks/blocks/patchembedding.py:32
↓ 15 callers
Class
RandRotate90d
Dictionary-based version :py:class:`monai.transforms.RandRotate90`. With probability `prob`, input arrays are rotated by 90 degrees in th
monai/transforms/spatial/dictionary.py:700
↓ 15 callers
Class
SimulateDelayd
Dictionary-based wrapper of :py:class:`monai.transforms.SimulateDelay`.
monai/transforms/utility/dictionary.py:879
↓ 15 callers
Class
Spacing
Resample input image into the specified `pixdim`. This transform is capable of lazy execution. See the :ref:`Lazy Resampling topic<lazy_resa
monai/transforms/spatial/array.py:341
↓ 15 callers
Class
UpCat
upsampling, concatenation with the encoder feature map, two convolutions
monai/networks/nets/basic_unet.py:91
↓ 15 callers
Class
WSIReader
Read whole slide images and extract patches using different backend libraries Args: backend: the name of backend whole slide image r
monai/data/wsi_reader.py:512
↓ 14 callers
Class
CheckpointSaver
CheckpointSaver acts as an Ignite handler to save checkpoint data into files. It supports to save according to metrics result, epoch number,
monai/handlers/checkpoint_saver.py:33
↓ 14 callers
Class
GeneralizedDiceLoss
Compute the generalised Dice loss defined in: Sudre, C. et. al. (2017) Generalised Dice overlap as a deep learning loss function
monai/losses/dice.py:311
↓ 14 callers
Class
GenerateHeatmap
Generate per-landmark Gaussian heatmaps for 2D or 3D coordinates. Notes: - Coordinates are interpreted in voxel units and expected i
monai/transforms/post/array.py:753
↓ 14 callers
Class
InvC
tests/transforms/test_random_order.py:32
↓ 14 callers
Class
InvD
tests/transforms/test_random_order.py:39
↓ 14 callers
Class
Resized
Dictionary-based wrapper of :py:class:`monai.transforms.Resize`. This transform is capable of lazy execution. See the :ref:`Lazy Resampling
monai/transforms/spatial/dictionary.py:786
↓ 14 callers
Class
SampleOperations
Apply statistical operation to a sample (image/ndarray/tensor). Notes: Percentile operation uses a partial function that embeds diff
monai/auto3dseg/operations.py:44
↓ 14 callers
Class
ScaleIntensity
Scale the intensity of input image to the given value range (minv, maxv). If `minv` and `maxv` not provided, use `factor` to scale image by `
monai/transforms/intensity/array.py:445
↓ 14 callers
Class
SupervisedEvaluator
Standard supervised evaluation method with image and label(optional), inherits from evaluator and Workflow. Args: device: an object
monai/engines/evaluator.py:174
↓ 14 callers
Class
SupervisedTrainer
Standard supervised training method with image and label, inherits from ``Trainer`` and ``Workflow``. Args: device: an object repres
monai/engines/trainer.py:80
↓ 14 callers
Class
ToNumpy
Converts the input data to numpy array, can support list or tuple of numbers and PyTorch Tensor. Args: dtype: target data type when
monai/transforms/utility/array.py:501
↓ 13 callers
Class
AutoencoderKlMaisi
AutoencoderKL with custom MaisiEncoder and MaisiDecoder. Args: spatial_dims: Number of spatial dimensions (1D, 2D, 3D). in_c
monai/apps/generation/maisi/networks/autoencoderkl_maisi.py:889
↓ 13 callers
Class
GDSDataset
An extension of the PersistentDataset using direct memory access(DMA) data path between GPU memory and storage, thus avoiding a bounce buffer
monai/data/dataset.py:1548
↓ 13 callers
Class
N
tests/transforms/test_compose_get_number_conversions.py:49
↓ 13 callers
Class
NumpyReader
Load NPY or NPZ format data based on Numpy library, they can be arrays or pickled objects. A typical usage is to load the `mask` data for cla
monai/data/image_reader.py:1224
↓ 13 callers
Class
RandFlipd
Dictionary-based version :py:class:`monai.transforms.RandFlip`. See `numpy.flip` for additional details. https://docs.scipy.org/doc/nump
monai/transforms/spatial/dictionary.py:1565
↓ 13 callers
Class
StatsHandler
StatsHandler defines a set of Ignite Event-handlers for all the log printing logics. It can be used for any Ignite Engine(trainer, validator
monai/handlers/stats_handler.py:36
↓ 13 callers
Class
T
tests/transforms/test_compose_get_number_conversions.py:55
↓ 13 callers
Class
ToTensord
Dictionary-based wrapper of :py:class:`monai.transforms.ToTensor`.
monai/transforms/utility/dictionary.py:442
↓ 13 callers
Class
TransformerBlock
A transformer block, based on: "Dosovitskiy et al., An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv
monai/networks/blocks/transformerblock.py:22
↓ 13 callers
Class
UnetrBasicBlock
A CNN module that can be used for UNETR, based on: "Hatamizadeh et al., UNETR: Transformers for 3D Medical Image Segmentation <https://arxiv.
monai/networks/blocks/unetr_block.py:209
↓ 12 callers
Class
CalibrationErrorMetric
Compute the Calibration Error between predicted probabilities and ground truth labels. **Why Calibration Matters:** A well-calibrated c
monai/metrics/calibration.py:162
↓ 12 callers
Class
CopyItemsd
Copy specified items from data dictionary and save with different key names. It can copy several items together and copy several times.
monai/transforms/utility/dictionary.py:909
↓ 12 callers
Class
DDIMScheduler
Denoising diffusion implicit models is a scheduler that extends the denoising procedure introduced in denoising diffusion probabilistic model
monai/networks/schedulers/ddim.py:43
↓ 12 callers
Class
DecoderOnlyTransformer
Decoder-only (Autoregressive) Transformer model. Args: num_tokens: Number of tokens in the vocabulary. max_seq_len: Maximum seque
monai/networks/nets/transformer.py:43
↓ 12 callers
Class
GaussianSmooth
Apply Gaussian smooth to the input data based on specified `sigma` parameter. A default value `sigma=1.0` is provided for reference. Arg
monai/transforms/intensity/array.py:1614
↓ 12 callers
Class
Lambda
Apply a user-defined lambda as a transform. For example: .. code-block:: python :emphasize-lines: 2 image = np.ones((1
monai/transforms/utility/array.py:783
↓ 12 callers
Class
NT
tests/transforms/test_compose_get_number_conversions.py:61
↓ 12 callers
Class
Orientation
Change the input image's orientation into the specified based on `axcodes`. This transform is capable of lazy execution. See the :ref:`Lazy
monai/transforms/spatial/array.py:553
↓ 12 callers
Class
RandAdjustContrast
Randomly changes image intensity with gamma transform. Each pixel/voxel intensity is updated as: x = ((x - min) / intensity_range) ^ gam
monai/transforms/intensity/array.py:1247
↓ 12 callers
Class
RandAffined
Dictionary-based wrapper of :py:class:`monai.transforms.RandAffine`. This transform is capable of lazy execution. See the :ref:`Lazy Resampl
monai/transforms/spatial/dictionary.py:1026
↓ 12 callers
Class
RandFlip
Randomly flips the image along axes. Preserves shape. See numpy.flip for additional details. https://docs.scipy.org/doc/numpy/reference/g
monai/transforms/spatial/array.py:1439
next →
1–100 of 2,334, ranked by callers