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