Package ai.konduit.serving.output.adapter

Class SSDOutputAdapter

  • All Implemented Interfaces:
    MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>
    public class SSDOutputAdapter
extends java.lang.Object
implements MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>
    An input adapter for ssd in tensorflow.
    Author:
    Adam Gibson

    • Constructor Summary

      Constructors 
      Constructor Description
      SSDOutputAdapter​(double threshold, int numLabels)  
      SSDOutputAdapter​(double threshold, java.io.InputStream labels, int numLabels)  
      SSDOutputAdapter​(double threshold, org.deeplearning4j.zoo.util.Labels labels, int numLabels)  

    • Method Summary

      All Methods Static Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      java.util.Map<java.lang.String,​BatchOutput> adapt​(org.nd4j.linalg.api.ndarray.INDArray[] input, java.util.List<java.lang.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​(java.io.InputStream is, int numLabels)  
      java.util.List<java.lang.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 java.lang.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,
                        java.io.InputStream labels,
                        int numLabels)

    • Method Detail

      • getLabels

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

      • getLabels

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

      • adapt

        public java.util.Map<java.lang.String,​BatchOutput> adapt​(org.nd4j.linalg.api.ndarray.INDArray[] input,
                                                               java.util.List<java.lang.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 java.util.List<java.lang.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