OnnxInferenceModel

open class OnnxInferenceModel : InferenceModel

Inference model built on ONNX format.

Since

0.3

Constructors

OnnxInferenceModel

fun OnnxInferenceModel()

Types

Companion

object Companion

Functions

open override fun close()

Releases the ONNXRuntime - related resources.

copy

open override fun copy(copiedModelName: String?, saveOptimizerState: Boolean, copyWeights: Boolean): TensorFlowInferenceModel

evaluate

fun evaluate(dataset: Dataset, metric: Metrics): Double

predict

open override fun predict(inputData: FloatArray): Int

fun predict(dataset: Dataset): List<Int>

fun predict(inputData: FloatArray, inputTensorName: String, outputTensorName: String): Int

Predicts the class of inputData.

predictRaw

fun predictRaw(inputData: FloatArray): Map<String, Any>

Returns list of multidimensional arrays with data from model outputs.

predictRawWithShapes

fun predictRawWithShapes(inputData: FloatArray): List<Pair<FloatBuffer, LongArray>>

Returns list of pairs from model outputs.

predictSoftly

fun predictSoftly(inputData: FloatArray): FloatArray

Predicts vector of probabilities instead of specific class in predict method.

open override fun predictSoftly(inputData: FloatArray, predictionTensorName: String): FloatArray

reshape

open override fun reshape(vararg dims: Long)

Chain-like setter to set up input shape.

toString

open override fun toString(): String

Properties

inputDataType

lateinit var inputDataType: OnnxJavaType

Data type for input tensor.

inputDimensions

open override val inputDimensions: LongArray

inputShape

lateinit var inputShape: LongArray

Data shape for prediction.

name

var name: String?

outputDataType

lateinit var outputDataType: OnnxJavaType

Data type for output tensor.

outputShape

lateinit var outputShape: LongArray

Data shape for prediction.

Inheritors

EfficientDetObjectDetectionModel

SSDMobileNetV1ObjectDetectionModel

SSDObjectDetectionModel

MultiPoseDetectionModel

SinglePoseDetectionModel