ai.djl.basicmodelzoo.cv.object_detection.ssd

Class SsdBlockFactory

java.lang.Object

ai.djl.basicmodelzoo.cv.object_detection.ssd.SsdBlockFactory

All Implemented Interfaces:

ai.djl.nn.BlockFactory, java.io.Serializable

public class SsdBlockFactory
extends java.lang.Object
implements ai.djl.nn.BlockFactory

A BlockFactory class that creates SingleShotDetection block.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SsdBlockFactory()

Method Summary

Modifier and Type	Method and Description
ai.djl.nn.Block	newBlock(ai.djl.Model model, java.nio.file.Path modelPath, java.util.Map<java.lang.String,?> arguments)

Methods inherited from class java.lang.Object

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

Constructor Detail

SsdBlockFactory

public SsdBlockFactory()

Method Detail

newBlock

public ai.djl.nn.Block newBlock(ai.djl.Model model,
                                java.nio.file.Path modelPath,
                                java.util.Map<java.lang.String,?> arguments)

Specified by:

newBlock in interface ai.djl.nn.BlockFactory