ai.djl.mxnet.zoo.cv.classification

Class Squeezenet

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader<java.awt.image.BufferedImage,ai.djl.modality.Classifications>

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

ai.djl.mxnet.zoo.cv.classification.Squeezenet

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader<java.awt.image.BufferedImage,ai.djl.modality.Classifications>

public class Squeezenet
extends ImageClassificationModelLoader

Model loader for Squeezenet Symbolic models.

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

See Also:

MxSymbolBlock

Field Summary

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

mrl, repository, version

Constructor Summary

Constructors

Constructor and Description
Squeezenet(ai.djl.repository.Repository repository) Creates a new instance of SqueezeNet.

Method Summary

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

getTranslator

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

listModels, loadModel, loadModel, match, search, 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, loadModel

Constructor Detail

Squeezenet

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

Creates a new instance of SqueezeNet.

Parameters:

repository - the repository to load the model from