Package org.jetbrains.kotlinx.dl.dataset.preprocessor

Types

ONNXModelPreprocessor

class ONNXModelPreprocessor(onnxModel: OnnxInferenceModel?, outputIndex: Int) : Preprocessor

Applies the given onnxModel as a preprocessing stage.

Transpose

class Transpose(axes: IntArray) : Preprocessor

Reverse or permute the axes of an input tensor.

Functions

onnx

fun TensorPreprocessing.onnx(block: ONNXModelPreprocessor.() -> Unit)

Image DSL Preprocessing extension.

transpose

fun TensorPreprocessing.transpose(sharpBlock: Transpose.() -> Unit)

Image DSL Preprocessing extension.