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

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

Bases: modelzoo.vision.pytorch.unet.input.UNetDataProcessor.UNetDataProcessor

Methods

create_dataloader

create_dataset

get_augment_transforms

get_tile_transform

preprocess_image

preprocess_mask

transform_image_and_mask

Attributes

tiling_image_shape

__init__(params)[source]#

previous

modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor

next

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