ai.konduit.serving.output.adapter

Class YOLOOutputAdapter

java.lang.Object

ai.konduit.serving.output.adapter.YOLOOutputAdapter

All Implemented Interfaces:

MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>

public class YOLOOutputAdapter
extends Object
implements MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>

Constructor Summary

Constructors

Constructor and Description
YOLOOutputAdapter(double threshold, InputStream labels, int numLabels)
YOLOOutputAdapter(double threshold, int numLabels)
YOLOOutputAdapter(double threshold, int[] inputShape, org.deeplearning4j.zoo.util.Labels labels, int numLabels, double[][] boundingBoxPriors)
YOLOOutputAdapter(double threshold, org.deeplearning4j.zoo.util.Labels labels, int numLabels)

Method Summary

Modifier and Type	Method and Description
Map<String,BatchOutput>	adapt(org.nd4j.linalg.api.ndarray.INDArray[] input, List<String> outputNames, io.vertx.ext.web.RoutingContext routingContext) Adapt a pair of INDArray and the output names, with the array input ordered by the output name
List<Class<? extends OutputAdapter<?>>>	outputAdapterTypes() Returns a map of the strings by output name to the OutputAdapter for each output

Methods inherited from class java.lang.Object

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

Constructor Detail

YOLOOutputAdapter

public YOLOOutputAdapter(double threshold,
                         int[] inputShape,
                         org.deeplearning4j.zoo.util.Labels labels,
                         int numLabels,
                         double[][] boundingBoxPriors)

YOLOOutputAdapter

public YOLOOutputAdapter(double threshold,
                         org.deeplearning4j.zoo.util.Labels labels,
                         int numLabels)

YOLOOutputAdapter

public YOLOOutputAdapter(double threshold,
                         int numLabels)

YOLOOutputAdapter

public YOLOOutputAdapter(double threshold,
                         InputStream labels,
                         int numLabels)

Method Detail

adapt

public Map<String,BatchOutput> adapt(org.nd4j.linalg.api.ndarray.INDArray[] input,
                                     List<String> outputNames,
                                     io.vertx.ext.web.RoutingContext routingContext)

Description copied from interface: MultiOutputAdapter

Adapt a pair of INDArray and the output names, with the array input ordered by the output name

Specified by:

adapt in interface MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>

Parameters:

input - the arrays to adapt

outputNames - the output names to adapt

routingContext - routing context

Returns:

Adapted inputs

outputAdapterTypes

public List<Class<? extends OutputAdapter<?>>> outputAdapterTypes()

Description copied from interface: MultiOutputAdapter

Returns a map of the strings by output name to the OutputAdapter for each output

Specified by:

outputAdapterTypes in interface MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>

Returns:

the output adapter types for this multi output adapter