modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor.CityscapesDataProcessor#
- class modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor.CityscapesDataProcessor[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