ai.djl.mxnet.zoo.cv.classification

Class Darknet

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.Classifications>

ai.djl.mxnet.zoo.cv.classification.ImageClassificationModelLoader

ai.djl.mxnet.zoo.cv.classification.Darknet

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.Classifications>

public class Darknet
extends ImageClassificationModelLoader

Model loader for Darknet Symbolic models.

The model was trained on Gluon and loaded in DJL in MXNet Symbol Block. See DarkNet.

See Also:

MxSymbolBlock

Field Summary

Fields inherited from class ai.djl.repository.zoo.BaseModelLoader

factories, modelZoo, mrl, repository, version

Constructor Summary

Constructors

Constructor and Description
Darknet(ai.djl.repository.Repository repository) Creates the Model loader from the given repository.

Method Summary

Methods inherited from class ai.djl.mxnet.zoo.cv.classification.ImageClassificationModelLoader

getApplication, loadModel

Methods inherited from class ai.djl.repository.zoo.BaseModelLoader

createModel, getArtifactId, listModels, loadModel, match, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.repository.zoo.ModelLoader

loadModel, loadModel, loadModel, loadModel

Constructor Detail

Darknet

public Darknet(ai.djl.repository.Repository repository)

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from