ObjectDetection

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

Object detection models and preprocessing.

Types

EfficientDetD0
Link copied to clipboard

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

EfficientDetD1
Link copied to clipboard

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

EfficientDetD2
Link copied to clipboard

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

EfficientDetD3
Link copied to clipboard

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

EfficientDetD4
Link copied to clipboard

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

EfficientDetD5
Link copied to clipboard

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

EfficientDetD6
Link copied to clipboard

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

SSD
Link copied to clipboard

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).

SSDMobileNetV1
Link copied to clipboard

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).

Functions

model
Link copied to clipboard
open fun model(modelHub: ModelHub): T
preInit
Link copied to clipboard
open fun preInit(): InferenceModel
preprocessInput
Link copied to clipboard
open fun preprocessInput(imageFile: File, preprocessing: Preprocessing): FloatArray
abstract fun preprocessInput(data: FloatArray, tensorShape: LongArray): FloatArray
pretrainedModel
Link copied to clipboard
abstract fun pretrainedModel(modelHub: ModelHub): U

Properties

channelsFirst
Link copied to clipboard
open override val channelsFirst: Boolean = true
modelRelativePath
Link copied to clipboard
open override val modelRelativePath: String

Inheritors

ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard
ONNXModels.ObjectDetection
Link copied to clipboard