org.deeplearning4j.zoo.model

Class ResNet50

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.ResNet50

All Implemented Interfaces:

InstantiableModel

public class ResNet50
extends ZooModel

Residual networks for deep learning. https://arxiv.org/abs/1512.03385

ImageNet weights for this model are available and have been converted from https://github.com/fchollet/keras/tree/1.1.2/keras/applications.

Author:

Justin Long (crockpotveggies)

Field Summary

Fields inherited from class org.deeplearning4j.zoo.ZooModel

ROOT_CACHE_DIR

Constructor Summary

Constructors

Constructor and Description
ResNet50(int numLabels, long seed, int iterations)
ResNet50(int outputNum, long seed, int iterations, WorkspaceMode workspaceMode)

Method Summary

Modifier and Type	Method and Description
ComputationGraphConfiguration.GraphBuilder	graphBuilder()
ComputationGraph	init()
ModelMetaData	metaData()
Class<? extends Model>	modelType()
long	pretrainedChecksum(PretrainedType pretrainedType)
String	pretrainedUrl(PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)
ZooType	zooType()

Methods inherited from class org.deeplearning4j.zoo.ZooModel

initPretrained, initPretrained, pretrainedAvailable

Methods inherited from class java.lang.Object

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

Constructor Detail

ResNet50

public ResNet50(int numLabels,
                long seed,
                int iterations)

ResNet50

public ResNet50(int outputNum,
                long seed,
                int iterations,
                WorkspaceMode workspaceMode)

Method Detail

pretrainedUrl

public String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

public long pretrainedChecksum(PretrainedType pretrainedType)

zooType

public ZooType zooType()

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)