cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset

class cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset

Bases: torchvision.datasets.VisionDataset

Methods

__init__(root, train_test_split, class_id_to_consider, split='train', transforms=None, transform=None, target_transform=None, use_worker_cache=False)

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

static __new__(cls, *args: Any, **kwargs: Any) → Any

previous

cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessor

next

cerebras.modelzoo.data.vision.segmentation.SkmDataProcessor