torch_em.transform

Data and label transformations for training neural networks.

View Source

1"""Data and label transformations for training neural networks.
2"""
3from .augmentation import get_augmentations
4from .defect import EMDefectAugmentation, get_artifact_source
5from .generic import Compose, Rescale, Tile, PadIfNecessary
6from .label import AffinityTransform, BoundaryTransform, NoToBackgroundBoundaryTransform, label_consecutive, labels_to_binary
7from .raw import get_raw_transform