Package ai.djl.modality.cv.translator

Class ObjectDetectionTranslatorFactory

java.lang.Object

ai.djl.translate.ExpansionTranslatorFactory<Image,O>

ai.djl.modality.cv.translator.BaseImageTranslatorFactory<DetectedObjects>

ai.djl.modality.cv.translator.ObjectDetectionTranslatorFactory

All Implemented Interfaces:

TranslatorFactory

Direct Known Subclasses:

InstanceSegmentationTranslatorFactory, SingleShotDetectionTranslatorFactory, YoloTranslatorFactory, YoloV5TranslatorFactory, YoloV8TranslatorFactory

public abstract class ObjectDetectionTranslatorFactory
extends BaseImageTranslatorFactory<DetectedObjects>

An abstract TranslatorFactory that creates a ObjectDetectionTranslator instance.

Nested Class Summary

Nested classes/interfaces inherited from class ai.djl.translate.ExpansionTranslatorFactory

ExpansionTranslatorFactory.TranslatorExpansion<IbaseT,ObaseT>

Constructor Summary

Constructors

Constructor	Description
ObjectDetectionTranslatorFactory()

Method Summary

Modifier and Type	Method	Description
Class<DetectedObjects>	getBaseOutputType()	Returns the output type for the base translator.

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslatorFactory

getBaseInputType, getExpansions, getPreprocessorExpansions

Methods inherited from class ai.djl.translate.ExpansionTranslatorFactory

buildBaseTranslator, getPostprocessorExpansions, getSupportedTypes, newInstance, withTranslator

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.translate.TranslatorFactory

isSupported

Constructor Details

ObjectDetectionTranslatorFactory

public ObjectDetectionTranslatorFactory()

Method Details

getBaseOutputType

public Class<DetectedObjects> getBaseOutputType()

Returns the output type for the base translator.

Specified by:

getBaseOutputType in class ExpansionTranslatorFactory<Image,DetectedObjects>

Returns:

the output type for the base translator