SSD

object SSD : ONNXModels.ObjectDetection<OnnxInferenceModel, SSDObjectDetectionModel>

This model is a real-time neural network for object detection that detects 80 different classes (labels are available in org.jetbrains.kotlinx.dl.dataset.handler.cocoCategoriesForSSD).

The model have an input with the 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

preInit

open override fun preInit(): SSDObjectDetectionModel

preprocessInput

open fun preprocessInput(imageFile: File, preprocessing: Preprocessing): FloatArray

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

pretrainedModel

open override fun pretrainedModel(modelHub: ModelHub): SSDObjectDetectionModel

Properties

channelsFirst

open override val channelsFirst: Boolean

modelRelativePath

open override val modelRelativePath: String