torch_em.transform
Data and label transformations for training neural networks.
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