ai.djl.mxnet.zoo.cv.classification

Class Alexnet

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.Alexnet

All Implemented Interfaces:

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

public class Alexnet
extends ImageClassificationModelLoader

Model loader for Alexnet Symbolic models.

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

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
Alexnet(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

Alexnet

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

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from