public class YoloTranslator extends ObjectDetectionTranslator
Modifier and Type | Class and Description |
---|---|
static class |
YoloTranslator.Builder
The builder for
YoloTranslator . |
ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ObjectDetectionTranslator.ObjectDetectionBuilder>
BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.ClassificationBuilder<T extends BaseImageTranslator.BaseBuilder>, BaseImageTranslator.SynsetLoader
classes, imageHeight, imageWidth, threshold
Constructor and Description |
---|
YoloTranslator(YoloTranslator.Builder builder)
Constructs an ImageTranslator with the provided builder.
|
Modifier and Type | Method and Description |
---|---|
static YoloTranslator.Builder |
builder()
Creates a builder to build a
YoloTranslator . |
static YoloTranslator.Builder |
builder(java.util.Map<java.lang.String,?> arguments)
Creates a builder to build a
YoloTranslator with specified arguments. |
DetectedObjects |
processOutput(TranslatorContext ctx,
NDList list)
Processes the output NDList to the corresponding output object.
|
prepare
getBatchifier, getBooleanValue, getFloatValue, getIntValue, getPipeline, getStringValue, processInput
public YoloTranslator(YoloTranslator.Builder builder)
builder
- the data to build withpublic DetectedObjects processOutput(TranslatorContext ctx, NDList list)
ctx
- the toolkit used for post-processinglist
- the output NDList after inferencepublic static YoloTranslator.Builder builder()
YoloTranslator
.public static YoloTranslator.Builder builder(java.util.Map<java.lang.String,?> arguments)
YoloTranslator
with specified arguments.arguments
- arguments to specify builder options