Preprocessing

class Preprocessing
Content copied to clipboard

The data preprocessing pipeline presented as Kotlin DSL on receivers.

Could be used to handle directory of images or one image file.

Constructors

Preprocessing
Link copied to clipboard
fun Preprocessing()
Content copied to clipboard

Functions

getFinalShape
Link copied to clipboard
fun getFinalShape(inputShape: ImageShape = ImageShape()): ImageShape
Content copied to clipboard

Returns the final shape of data when image preprocessing is applied to the image with the given shape.

invoke
Link copied to clipboard
operator fun invoke(imagePath: File): Pair<FloatArray, ImageShape>
Content copied to clipboard

Applies the preprocessing pipeline to the specific image file.

Properties

imagePreprocessingStage
Link copied to clipboard
lateinit var imagePreprocessingStage: ImagePreprocessing
Content copied to clipboard

This stage describes the process of image loading and transformation before converting to tensor.

tensorPreprocessingStage
Link copied to clipboard
lateinit var tensorPreprocessingStage: TensorPreprocessing
Content copied to clipboard

This stage describes the process of data transformation after converting to tensor.

Extensions

transformImage
Link copied to clipboard
fun Preprocessing.transformImage(block: ImagePreprocessing.() -> Unit)
Content copied to clipboard

Defines preprocessing operations on the image.

transformTensor
Link copied to clipboard
fun Preprocessing.transformTensor(block: TensorPreprocessing.() -> Unit)
Content copied to clipboard

Defines preprocessing operations on the tensor.