SSD

object SSD : ONNXModels.ObjectDetection<OnnxInferenceModel, SSDObjectDetectionModel>

This model is a real-time neural network for object detection that detects 80 different classes.

Image shape is (1x3x1200x1200).

The model has 3 outputs:

boxes: (1x'nbox'x4)
labels: (1x'nbox')
scores: (1x'nbox')

See also

Detailed description of SSD model and its pre- and postprocessing in onnx/models repository.

Functions

model

open fun model(modelHub: ModelHub): OnnxInferenceModel

preprocessInput

open fun preprocessInput(preprocessing: Preprocessing): FloatArray

open override fun preprocessInput(data: FloatArray, tensorShape: LongArray): FloatArray

pretrainedModel

open override fun pretrainedModel(modelHub: ModelHub): SSDObjectDetectionModel

Properties

modelRelativePath

open override val modelRelativePath: String