mipcandy.data.transform

Module Contents

Classes

JointTransform
MONAITransform

API

class mipcandy.data.transform.JointTransform(*, transform: mipcandy.types.Transform | None = None, image_only: mipcandy.types.Transform | None = None, label_only: mipcandy.types.Transform | None = None, keys: tuple[str, str] = ('image', 'label'), order: tuple[mipcandy.data.transform._Order, mipcandy.data.transform._Order, mipcandy.data.transform._Order] = ('transform', 'image_only', 'label_only'))

Bases: torch.nn.Module

forward(image: torch.Tensor, label: torch.Tensor) → tuple[torch.Tensor, torch.Tensor]

class mipcandy.data.transform.MONAITransform(transform: mipcandy.types.Transform, *, keys: tuple[str, str] = ('image', 'label'))

Bases: torch.nn.Module

forward(data: torch.Tensor | dict[str, torch.Tensor]) → torch.Tensor | dict[str, torch.Tensor]