ai.djl.modality.cv

Class DetectedObjects

java.lang.Object

ai.djl.modality.Classifications

ai.djl.modality.cv.DetectedObjects

public class DetectedObjects
extends Classifications

A class representing the detected objects results for a single image in an object detection case.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DetectedObjects.DetectedObject A DetectedObject represents a single potential detected Object for an image.

Nested classes/interfaces inherited from class ai.djl.modality.Classifications

Classifications.Classification

Field Summary

Fields inherited from class ai.djl.modality.Classifications

classNames, probabilities

Constructor Summary

Constructors

Constructor and Description
DetectedObjects(java.util.List<java.lang.String> classNames, java.util.List<java.lang.Double> probabilities, java.util.List<BoundingBox> boundingBoxes) Constructs a DetectedObjects, usually during post-processing.

Method Summary

Modifier and Type	Method and Description
int	getNumberOfObjects() Returns the number of objects found in an image.
<T extends Classifications.Classification> T	item(int index) Returns the item at a given index based on the order used to construct the Classifications.

Methods inherited from class ai.djl.modality.Classifications

best, get, items, topK, toString

Methods inherited from class java.lang.Object

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

Constructor Detail

DetectedObjects

public DetectedObjects(java.util.List<java.lang.String> classNames,
                       java.util.List<java.lang.Double> probabilities,
                       java.util.List<BoundingBox> boundingBoxes)

Constructs a DetectedObjects, usually during post-processing.

All three inputs(classNames, probabilities, boundingBoxes) should be parallel lists.

Parameters:

classNames - the names of the objects that were detected

probabilities - the probability of the objects that were detected

boundingBoxes - the bounding boxes of the objects that were detected

Method Detail

item

public <T extends Classifications.Classification> T item(int index)

Returns the item at a given index based on the order used to construct the Classifications.

Overrides:

item in class Classifications

Type Parameters:

T - the type of classification item for the task

Parameters:

index - the index of the item to return

Returns:

the item at the given index, equivalent to classifications.items().get(index)

getNumberOfObjects

public int getNumberOfObjects()

Returns the number of objects found in an image.

Returns:

the number of objects found in an image