modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset#

class modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset[source]#

Bases: torchvision.datasets.VisionDataset

Methods

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

Call self as a function.

__init__(root, train_test_split, class_id_to_consider, split='train', transforms=None, transform=None, target_transform=None, use_worker_cache=False)[source]#
static __new__(cls, *args: Any, **kwargs: Any) Any#