ai.konduit.serving.output.adapter

Class SSDOutputAdapter

java.lang.Object

ai.konduit.serving.output.adapter.SSDOutputAdapter

All Implemented Interfaces:

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

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

An input adapter for ssd in tensorflow.

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	DEFAULT_LABELS_RESOURCE_NAME

Constructor Summary

Constructors

Constructor and Description
SSDOutputAdapter(double threshold, InputStream labels, int numLabels)
SSDOutputAdapter(double threshold, int numLabels)
SSDOutputAdapter(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
static org.deeplearning4j.zoo.util.Labels	getLabels()
static org.deeplearning4j.zoo.util.Labels	getLabels(InputStream is, int numLabels)
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

Field Detail

DEFAULT_LABELS_RESOURCE_NAME

public static final String DEFAULT_LABELS_RESOURCE_NAME

See Also:

Constant Field Values

Constructor Detail

SSDOutputAdapter

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

SSDOutputAdapter

public SSDOutputAdapter(double threshold,
                        int numLabels)

SSDOutputAdapter

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

Method Detail

getLabels

public static org.deeplearning4j.zoo.util.Labels getLabels(InputStream is,
                                                           int numLabels)

getLabels

public static org.deeplearning4j.zoo.util.Labels getLabels()

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