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

Types

ONNXModelPreprocessor
Link copied to clipboard
class ONNXModelPreprocessor(onnxModel: OnnxInferenceModel?, outputIndex: Int) : Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Applies the given onnxModel as a preprocessing stage.

Transpose
Link copied to clipboard
class Transpose(axes: IntArray) : Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>

Reverse or permute the axes of an input tensor.

Functions

onnx
Link copied to clipboard
fun <I> Operation<I, Pair<FloatArray, TensorShape>>.onnx(block: ONNXModelPreprocessor.() -> Unit): Operation<I, Pair<FloatArray, TensorShape>>

Image DSL Preprocessing extension.

transpose
Link copied to clipboard
fun <I> Operation<I, Pair<FloatArray, TensorShape>>.transpose(sharpBlock: Transpose.() -> Unit): Operation<I, Pair<FloatArray, TensorShape>>

The DSL extension function for Transpose operation.