ai.djl.mxnet.zoo.cv.classification

Class MlpModelLoader

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

All Implemented Interfaces:

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

public class MlpModelLoader
extends ImageClassificationModelLoader

Model loader for MLP Symbolic models.

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

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

MlpModelLoader

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

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from