ai.djl.basicmodelzoo.cv.object_detection.ssd

Class SingleShotDetection

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.basicmodelzoo.cv.object_detection.ssd.SingleShotDetection

All Implemented Interfaces:

ai.djl.nn.Block

public final class SingleShotDetection
extends ai.djl.nn.AbstractBlock

SingleShotDetection is an implementation of Block that implements a Single Shot Detection (SSD) model for object detection.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SingleShotDetection.Builder The Builder to construct a SingleShotDetection.

Field Summary

Fields inherited from class ai.djl.nn.AbstractBlock

inputNames, inputShapes

Method Summary

Modifier and Type	Method and Description
static SingleShotDetection.Builder	builder() Creates a builder to build a SingleShotDetection.
ai.djl.ndarray.NDList	forward(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, ai.djl.util.PairList<java.lang.String,java.lang.Object> params)
static ai.djl.nn.convolutional.Conv2D	getAnchorPredictionBlock(int numAnchors) Creates a anchor prediction block used in an SSD.
ai.djl.nn.BlockList	getChildren()
static ai.djl.nn.convolutional.Conv2D	getClassPredictionBlock(int numAnchors, int numClasses) Creates a class prediction block used in an SSD.
java.util.List<ai.djl.nn.Parameter>	getDirectParameters()
static ai.djl.nn.SequentialBlock	getDownSamplingBlock(int numFilters) Creates a Block that reduces the size of a convolutional block by half.
ai.djl.ndarray.types.Shape[]	getOutputShapes(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.Shape[] inputShapes)
ai.djl.ndarray.types.Shape	getParameterShape(java.lang.String name, ai.djl.ndarray.types.Shape[] inputShapes)
ai.djl.ndarray.types.Shape[]	initialize(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.types.DataType dataType, ai.djl.ndarray.types.Shape... inputShapes)
void	loadParameters(ai.djl.ndarray.NDManager manager, java.io.DataInputStream is)
void	saveParameters(java.io.DataOutputStream os)

Methods inherited from class ai.djl.nn.AbstractBlock

beforeInitialize, cast, clear, describeInput, getParameters, isInitialized, readInputShapes, saveInputShapes, setInitializer, setInitializer

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.nn.Block

forward, validateLayout

Method Detail

forward

public ai.djl.ndarray.NDList forward(ai.djl.training.ParameterStore parameterStore,
                                     ai.djl.ndarray.NDList inputs,
                                     ai.djl.util.PairList<java.lang.String,java.lang.Object> params)

getDirectParameters

public java.util.List<ai.djl.nn.Parameter> getDirectParameters()

getParameterShape

public ai.djl.ndarray.types.Shape getParameterShape(java.lang.String name,
                                                    ai.djl.ndarray.types.Shape[] inputShapes)

getOutputShapes

public ai.djl.ndarray.types.Shape[] getOutputShapes(ai.djl.ndarray.NDManager manager,
                                                    ai.djl.ndarray.types.Shape[] inputShapes)

initialize

public ai.djl.ndarray.types.Shape[] initialize(ai.djl.ndarray.NDManager manager,
                                               ai.djl.ndarray.types.DataType dataType,
                                               ai.djl.ndarray.types.Shape... inputShapes)

getChildren

public ai.djl.nn.BlockList getChildren()

saveParameters

public void saveParameters(java.io.DataOutputStream os)
                    throws java.io.IOException

Throws:

java.io.IOException

loadParameters

public void loadParameters(ai.djl.ndarray.NDManager manager,
                           java.io.DataInputStream is)
                    throws java.io.IOException,
                           ai.djl.MalformedModelException

Throws:

java.io.IOException

ai.djl.MalformedModelException

getDownSamplingBlock

public static ai.djl.nn.SequentialBlock getDownSamplingBlock(int numFilters)

Creates a Block that reduces the size of a convolutional block by half.

Parameters:

numFilters - the number of filters

Returns:

a Block that reduces the size of a convolutional block by half

getClassPredictionBlock

public static ai.djl.nn.convolutional.Conv2D getClassPredictionBlock(int numAnchors,
                                                                     int numClasses)

Creates a class prediction block used in an SSD.

Parameters:

numAnchors - the number of anchors

numClasses - the number of classes

Returns:

a class prediction block used in an SSD

getAnchorPredictionBlock

public static ai.djl.nn.convolutional.Conv2D getAnchorPredictionBlock(int numAnchors)

Creates a anchor prediction block used in an SSD.

Parameters:

numAnchors - the number of anchors

Returns:

a anchor prediction block used in an SSD

builder

public static SingleShotDetection.Builder builder()

Creates a builder to build a SingleShotDetection.

Returns:

a new builder