ObjectDetection

sealed class ObjectDetection<T : InferenceModel, U : InferenceModel> : ModelType<T, U>

Object detection models and preprocessing.

Types

SSD

object SSD : ONNXModels.ObjectDetection<OnnxInferenceModel, SSDObjectDetectionModel>

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

YOLOv4

object YOLOv4 : ONNXModels.ObjectDetection<OnnxInferenceModel, OnnxInferenceModel>

Functions

model

open fun model(modelHub: ModelHub): T

preprocessInput

open fun preprocessInput(preprocessing: Preprocessing): FloatArray

abstract fun preprocessInput(data: FloatArray, tensorShape: LongArray): FloatArray

pretrainedModel

abstract fun pretrainedModel(modelHub: ModelHub): U

Properties

modelRelativePath

open override val modelRelativePath: String

Inheritors

ONNXModels.ObjectDetection

ONNXModels.ObjectDetection