org.deeplearning4j.zoo.model

Class SqueezeNet

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.SqueezeNet

All Implemented Interfaces:

InstantiableModel

public class SqueezeNet
extends ZooModel

U-Net An implementation of SqueezeNet. Touts similar accuracy to AlexNet with a fraction of the parameters.

Paper: https://arxiv.org/abs/1602.07360

ImageNet weights for this model are available and have been converted from https://github.com/rcmalli/keras-squeezenet/.

Author:

Justin Long (crockpotveggies)

Method Summary

Modifier and Type	Method and Description
ComputationGraphConfiguration.GraphBuilder	graphBuilder()
ComputationGraph	init()
ComputationGraph	initPretrained(PretrainedType pretrainedType) Returns a pretrained model for the given dataset, if available.
ModelMetaData	metaData()
Class<? extends Model>	modelType()
long	pretrainedChecksum(PretrainedType pretrainedType)
String	pretrainedUrl(PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)

Methods inherited from class org.deeplearning4j.zoo.ZooModel

initPretrained, modelName, pretrainedAvailable

Methods inherited from class java.lang.Object

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

Method Detail

pretrainedUrl

public String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

public long pretrainedChecksum(PretrainedType pretrainedType)

initPretrained

public ComputationGraph initPretrained(PretrainedType pretrainedType)
                                throws IOException

Description copied from class: ZooModel

Returns a pretrained model for the given dataset, if available.

Overrides:

initPretrained in class ZooModel

Returns:

Throws:

IOException

modelType

public Class<? extends Model> modelType()

init

public ComputationGraph init()

graphBuilder

public ComputationGraphConfiguration.GraphBuilder graphBuilder()

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)