SSDLikeModel

class SSDLikeModel(internalModel: OnnxInferenceModel, metadata: SSDLikeModelMetadata, modelKindDescription: String?) : SSDLikeModelBase<Bitmap> , CameraXCompatibleModel, InferenceModel

Special model class for detection objects on images with built-in preprocessing and post-processing. Suitable for models with SSD like output decoding.

It internally uses ONNXModels.ObjectDetection.EfficientDetLite0 or other SSDLike models trained on the COCO dataset.

Since

0.5

Parameters

internalModel

model used to make predictions

Constructors

SSDLikeModel

android

fun SSDLikeModel(internalModel: OnnxInferenceModel, metadata: SSDLikeModelMetadata, modelKindDescription: String? = null)

Functions

android

open override fun close()

convert

android

open override fun convert(output: OrtSession.Result): List<DetectedObject>

Converts raw model output to the result.

copy

android

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

detectObjects

android

fun detectObjects(image: Bitmap, topK: Int = 5): List<DetectedObject>

Returns the detected object for the given image sorted by the score.

initializeWith

android

open override fun initializeWith(vararg executionProviders: ExecutionProvider)

Initialize the model with the specified executions providers.

predict

android

open override fun predict(inputData: FloatArray): Int

open fun predict(input: Bitmap): List<DetectedObject>

Makes prediction on the given input.

predictSoftly

android

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

reshape

android

open override fun reshape(vararg dims: Long)

summary

android

open override fun summary(): ModelSummary

Properties

inputDimensions

android

open override val inputDimensions: LongArray

internalModel

android

open override val internalModel: OnnxInferenceModel

modelKindDescription

android

open override val modelKindDescription: String?

High-level description of the model. Used for model summary printing. For the model hub models it equals to the string representation of the OnnxModelType

preprocessing

android

open override val preprocessing: Operation<Bitmap, Pair<FloatArray, TensorShape>>

Preprocessing operation specific to this model.

targetRotation

android

open override var targetRotation: Int = 0

Target image rotation.