Package ai.konduit.serving.output.adapter

Class YOLOOutputAdapter

  • All Implemented Interfaces:
    MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>
    public class YOLOOutputAdapter
extends java.lang.Object
implements MultiOutputAdapter<org.nd4j.linalg.api.ndarray.INDArray[]>

    • Constructor Summary

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

    • Method Summary

      All 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
      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

    • 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,
                         java.io.InputStream labels,
                         int numLabels)

    • Method Detail

      • 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