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 #