Package ai.djl.modality.cv.translator
Class YoloTranslatorFactory
java.lang.Object
ai.djl.translate.ExpansionTranslatorFactory<Image,O>
ai.djl.modality.cv.translator.BaseImageTranslatorFactory<DetectedObjects>
ai.djl.modality.cv.translator.ObjectDetectionTranslatorFactory
ai.djl.modality.cv.translator.YoloTranslatorFactory
- All Implemented Interfaces:
TranslatorFactory
,Serializable
An
TranslatorFactory
that creates a YoloTranslator
instance.- See Also:
-
Nested Class Summary
Nested classes/interfaces inherited from class ai.djl.translate.ExpansionTranslatorFactory
ExpansionTranslatorFactory.TranslatorExpansion<IbaseT,
ObaseT> -
Constructor Summary
-
Method Summary
Modifier and TypeMethodDescriptionprotected Translator<Image,
DetectedObjects> buildBaseTranslator
(Model model, Map<String, ?> arguments) Builds the base translator that can be expanded.Methods inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslatorFactory
getBaseOutputType
Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslatorFactory
getBaseInputType, getExpansions, getPreprocessorExpansions
Methods inherited from class ai.djl.translate.ExpansionTranslatorFactory
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
-
YoloTranslatorFactory
public YoloTranslatorFactory()
-
-
Method Details
-
buildBaseTranslator
protected Translator<Image,DetectedObjects> buildBaseTranslator(Model model, Map<String, ?> arguments) Builds the base translator that can be expanded.- Specified by:
buildBaseTranslator
in classExpansionTranslatorFactory<Image,
DetectedObjects> - Parameters:
model
- theModel
that uses theTranslator
arguments
- the configurations for a newTranslator
instance- Returns:
- a base translator that can be expanded to form the factory options
-