ImageSegmentation

			ImageSegmentation(size, classes; kwargs...)

Learning task for image segmentation. Images are resized to size and a class is predicted for every pixel.

Keyword arguments

• computestats = false: Whether to compute image statistics on dataset data or use default ImageNet stats.

• aug_projections = DataAugmentation.Identity: augmentation to apply during ProjectiveTransforms (resizing and cropping)

• aug_image = DataAugmentation.Identity: pixel-level augmentation to apply during ImagePreprocessing

• C = RGB{N0f8}: Color type images are converted to before further processing. Use Gray{N0f8} for grayscale images.