torch_em.data
Implementations of Datasets and Samplers for training neural networks with PyTorch.
1"""Implementations of Datasets and Samplers for training neural networks with PyTorch. 2""" 3from .concat_dataset import ConcatDataset 4from .dataset_wrapper import DatasetWrapper 5from .image_collection_dataset import ImageCollectionDataset 6from .raw_dataset import RawDataset 7from .pseudo_label_dataset import PseudoLabelDataset 8from .raw_image_collection_dataset import RawImageCollectionDataset 9from .segmentation_dataset import SegmentationDataset 10from .sampler import ( 11 MinForegroundSampler, 12 MinInstanceSampler, 13 MinIntensitySampler, 14 MinNoToBackgroundBoundarySampler, 15 MinTwoInstanceSampler, 16 MinSemanticLabelForegroundSampler, 17)