Package ai.djl.modality.cv.translator

Class YoloV5TranslatorFactory

java.lang.Object

ai.djl.modality.cv.translator.ObjectDetectionTranslatorFactory

ai.djl.modality.cv.translator.YoloV5TranslatorFactory

All Implemented Interfaces:

TranslatorFactory

public class YoloV5TranslatorFactory
extends ObjectDetectionTranslatorFactory

An TranslatorFactory that creates a YoloV5Translator instance.

Constructor Summary

Constructors

Constructor	Description
YoloV5TranslatorFactory()

Method Summary

Modifier and Type	Method	Description
Translator<?,?>	newInstance(java.lang.Class<?> input, java.lang.Class<?> output, Model model, java.util.Map<java.lang.String,?> arguments)	Returns a new instance of the Translator class.

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

getSupportedTypes

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 Detail

YoloV5TranslatorFactory

public YoloV5TranslatorFactory()

Method Detail

newInstance

public Translator<?,?> newInstance(java.lang.Class<?> input,
                                         java.lang.Class<?> output,
                                         Model model,
                                         java.util.Map<java.lang.String,?> arguments)
                                  throws TranslateException

Returns a new instance of the Translator class.

Parameters:

input - the input class

output - the output class

model - the Model that uses the Translator

arguments - the configurations for a new Translator instance

Returns:

a new instance of the Translator class

Throws:

TranslateException - if failed to create Translator instance