ai.djl.training.evaluator

Class SingleShotDetectionAccuracy

java.lang.Object

ai.djl.training.evaluator.Evaluator

ai.djl.training.evaluator.AbstractAccuracy

ai.djl.training.evaluator.SingleShotDetectionAccuracy

public class SingleShotDetectionAccuracy
extends AbstractAccuracy

SingleShotDetectionAccuracy is an implementation of AbstractAccuracy. It is used while training a Single Shot Detection (SSD) model for object detection. It uses the targets computed by MultiBoxTarget, and computes the class prediction accuracy against the computed targets.

Field Summary

Fields inherited from class ai.djl.training.evaluator.AbstractAccuracy

axis, correctInstances, index

Fields inherited from class ai.djl.training.evaluator.Evaluator

totalInstances

Constructor Summary

Constructors

Constructor and Description
SingleShotDetectionAccuracy(java.lang.String name) Creates a new instance of SingleShotDetectionAccuracy with the given name.

Method Summary

Modifier and Type	Method and Description
protected ai.djl.util.Pair<java.lang.Long,NDArray>	accuracyHelper(NDList labels, NDList predictions) A helper for classes extending AbstractAccuracy.

Methods inherited from class ai.djl.training.evaluator.AbstractAccuracy

addAccumulator, evaluate, getAccumulator, resetAccumulator, updateAccumulator

Methods inherited from class ai.djl.training.evaluator.Evaluator

checkLabelShapes, checkLabelShapes, getName

Methods inherited from class java.lang.Object

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

Constructor Detail

SingleShotDetectionAccuracy

public SingleShotDetectionAccuracy(java.lang.String name)

Creates a new instance of SingleShotDetectionAccuracy with the given name.

Parameters:

name - the name given to the accuracy

Method Detail

accuracyHelper

protected ai.djl.util.Pair<java.lang.Long,NDArray> accuracyHelper(NDList labels,
                                                                  NDList predictions)

Description copied from class: AbstractAccuracy

A helper for classes extending AbstractAccuracy.

Specified by:

accuracyHelper in class AbstractAccuracy

Parameters:

labels - the labels to get accuracy for

predictions - the predictions to get accuracy for

Returns:

a pair(number of total values, ndarray int of correct values)