SSDObjectDetectionModel

class SSDObjectDetectionModel : OnnxInferenceModel

Special model class for detection objects on images with built-in preprocessing and post-processing.

It internally uses SSD model trained on the COCO dataset.

Since

0.3

Constructors

SSDObjectDetectionModel

fun SSDObjectDetectionModel()

Functions

open override fun close()

copy

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

detectObjects

fun detectObjects(imageFile: File, topK: Int = 5): List<DetectedObject>

fun detectObjects(inputData: FloatArray, topK: Int = 5): List<DetectedObject>

Returns the top N detected object for the given image file.

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): List<Array<*>>

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

inputDimensions

open override val inputDimensions: LongArray

inputShape

lateinit var inputShape: LongArray

Data shape for prediction.

name

var name: String?

outputShape

lateinit var outputShape: LongArray

Data shape for prediction.

SSDObjectDetectionModel

Since

See also

Constructors

Functions

Properties