org.apache.mxnet.infer

ObjectDetector

Related Doc: package infer

class ObjectDetector extends AnyRef

The ObjectDetector class helps to run ObjectDetection tasks where the goal is to find bounding boxes and corresponding labels for objects in a image.

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. ObjectDetector
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ObjectDetector(modelPathPrefix: String, inputDescriptors: IndexedSeq[DataDesc], contexts: Array[Context] = Context.cpu(), epoch: Option[Int] = Some(0))

    modelPathPrefix

    Path prefix from where to load the model artifacts. These include the symbol, parameters, and synset.txt. Example: file://model-dir/ssd_resnet50_512 (containing ssd_resnet50_512-symbol.json, ssd_resnet50_512-0000.params, and synset.txt)

    inputDescriptors

    Descriptors defining the input node names, shape, layout and type parameters

    contexts

    Device contexts on which you want to run inference. Defaults to CPU.

    epoch

    Model epoch to load; defaults to 0

Value Members

  1. final def !=(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  5. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  6. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  10. val handler: MXNetHandler

    Attributes
    protected[org.apache.mxnet.infer]
  11. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  12. val height: Int

    Attributes
    protected[org.apache.mxnet.infer]
  13. def imageBatchObjectDetect(inputBatch: Traversable[BufferedImage], topK: Option[Int] = None): IndexedSeq[IndexedSeq[(String, Array[Float])]]

    To classify batch of input images according to the provided model

    To classify batch of input images according to the provided model

    inputBatch

    Input array of buffered images

    topK

    Number of result elements to return, sorted by probability

    returns

    List of list of tuples of (class, probability)

  14. def imageObjectDetect(inputImage: BufferedImage, topK: Option[Int] = None): IndexedSeq[IndexedSeq[(String, Array[Float])]]

    Detects objects and returns bounding boxes with corresponding class/label

    Detects objects and returns bounding boxes with corresponding class/label

    inputImage

    Path prefix of the input image

    topK

    Number of result elements to return, sorted by probability

    returns

    List of list of tuples of (class, [probability, xmin, ymin, xmax, ymax])

  15. val imgClassifier: ImageClassifier

    Attributes
    protected[org.apache.mxnet.infer]
  16. val inputShape: Shape

    Attributes
    protected[org.apache.mxnet.infer]
  17. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  18. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  19. final def notify(): Unit

    Definition Classes
    AnyRef
  20. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  21. def objectDetectWithNDArray(input: IndexedSeq[NDArray], topK: Option[Int]): IndexedSeq[IndexedSeq[(String, Array[Float])]]

    Takes input images as NDArrays.

    Takes input images as NDArrays. Useful when you want to perform multiple operations on the input array, or when you want to pass a batch of input images.

    input

    Indexed Sequence of NDArrays

    topK

    (Optional) How many top_k (sorting will be based on the last axis) elements to return. If not passed, returns all unsorted output.

    returns

    List of list of tuples of (class, [probability, xmin, ymin, xmax, ymax])

  22. val predictor: PredictBase

    Attributes
    protected[org.apache.mxnet.infer]
  23. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  24. val synset: IndexedSeq[String]

    Attributes
    protected[org.apache.mxnet.infer]
  25. def toString(): String

    Definition Classes
    AnyRef → Any
  26. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  27. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  28. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  29. val width: Int

    Attributes
    protected[org.apache.mxnet.infer]

Inherited from AnyRef

Inherited from Any

Ungrouped