Package org.jetbrains.kotlinx.dl.impl.preprocessing.image

Types

ArrayTransform

jvm

This interface represents a custom transformation of a FloatArray.

CenterCrop

jvm

class CenterCrop(size: Int) : Operation<BufferedImage, BufferedImage>

This image preprocessor defines centerCrop operation. It crops the given image at the center. If the image size is smaller than the output size along any edge, the image is padded with 0 and then center cropped.

ColorMode

jvm

enum ColorMode : Enum<ColorMode>

Represents the number and order of color channels in the image.

Convert

jvm

class Convert(colorMode: ColorMode) : Operation<BufferedImage, BufferedImage>

Converts provided image to the desired ColorMode.

ConvertToFloatArray

jvm

class ConvertToFloatArray : Operation<BufferedImage, Pair<FloatArray, TensorShape>>

Converts BufferedImage to float array representation.

Cropping

jvm

class Cropping(top: Int, bottom: Int, left: Int, right: Int) : Operation<BufferedImage, BufferedImage>

This image preprocessor defines the Crop operation.

ImageConverter

jvm

object ImageConverter

Helper object with methods to convert BufferedImage to FloatArray.

InterpolationType

jvm

enum InterpolationType : Enum<InterpolationType>

When an image is resized, we need an interpolation algorithm to draw newly created pixels between old pixels.

Padding

jvm

class Padding(top: Int, bottom: Int, left: Int, right: Int, mode: PaddingMode) : Operation<BufferedImage, BufferedImage>

This image preprocessor defines the Pad operation.

PaddingMode

jvm

interface PaddingMode

Type of padding to use.

RenderingSpeed

jvm

enum RenderingSpeed : Enum<RenderingSpeed>

The speed of single file preprocessing could be tuned via this setting.

Resize

jvm

class Resize(outputWidth: Int, outputHeight: Int, interpolation: InterpolationType, renderingSpeed: RenderingSpeed, enableAntialiasing: Boolean) : Operation<BufferedImage, BufferedImage>

This image preprocessor defines the Resize operation.

Rotate

jvm

class Rotate(degrees: Float, interpolation: InterpolationType, renderingSpeed: RenderingSpeed, enableAntialiasing: Boolean) : Operation<BufferedImage, BufferedImage>

This image preprocessor defines the Rotate operation.

Functions

argB8888ToNCHWArray

common

fun argB8888ToNCHWArray(encodedPixels: IntArray, width: Int, height: Int, channels: Int): FloatArray

Decodes an ARGB8888 encoded pixel array to a float array containing the red, green, blue components in NCWH layout.

argB8888ToNHWCArray

common

fun argB8888ToNHWCArray(encodedPixels: IntArray, width: Int, height: Int, channels: Int): FloatArray

Decodes an ARGB8888 encoded pixel array to a float array containing the red, green, blue components in NHWC layout.

centerCrop

jvm

fun Operation<I, BufferedImage>.centerCrop(block: CenterCrop.() -> Unit): Operation<I, BufferedImage>

Applies CenterCrop operation to crop the image at the center.

convert

jvm

fun Operation<I, BufferedImage>.convert(block: Convert.() -> Unit): Operation<I, BufferedImage>

Applies Convert operation to convert the image to a different ColorMode.

crop

jvm

fun Operation<I, BufferedImage>.crop(block: Cropping.() -> Unit): Operation<I, BufferedImage>

Applies Cropping operation to crop the image by the specified amount.

decodeARGB8888Pixel

common

fun decodeARGB8888Pixel(pixelValue: Int): <ERROR CLASS><Float, Float, Float>

Decodes an ARGB8888 encoded pixel to a red, green, blue components.

denormalizeInplace

jvm

fun denormalizeInplace(input: FloatArray, scale: Float): FloatArray

Each array element of an input is multiplied in-place by scale coefficient.

grayscale

jvm

fun Operation<I, BufferedImage>.grayscale(): Operation<I, BufferedImage>

Applies Convert operation to convert the image to ColorMode.GRAYSCALE.

imageType

jvm

fun ColorMode.imageType(): Int

Returns an integer representing a type of BufferedImage corresponding to this color mode.

pad

jvm

fun Operation<I, BufferedImage>.pad(block: Padding.() -> Unit): Operation<I, BufferedImage>

Applies Padding operation to pad the image.

resize

jvm

fun Operation<I, BufferedImage>.resize(block: Resize.() -> Unit): Operation<I, BufferedImage>

Applies Resize operation to resize the image to a specific size.

rotate

jvm

fun Operation<I, BufferedImage>.rotate(block: Rotate.() -> Unit): Operation<I, BufferedImage>

Applies Rotate operation to rotate the image by an arbitrary angle (specified in degrees).

toFloatArray

jvm

fun Operation<I, BufferedImage>.toFloatArray(block: ConvertToFloatArray.() -> Unit): Operation<I, Pair<FloatArray, TensorShape>>

Applies ConvertToFloatArray operation to convert the image to a float array.