org.deeplearning4j.zoo.model

Class VGG16

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.VGG16

All Implemented Interfaces:

InstantiableModel

public class VGG16
extends ZooModel

VGG-16, from Very Deep Convolutional Networks for Large-Scale Image Recognition https://arxiv.org/abs/1409.1556

Deep Face Recognition
http://www.robots.ox.ac.uk/~vgg/publications/2015/Parkhi15/parkhi15.pdf

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

CIFAR-10 weights for this model are available and have been converted using "approach 2" from https://github.com/rajatvikramsingh/cifar10-vgg16.

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

Author:

Justin Long (crockpotveggies)

Method Summary

Modifier and Type	Method and Description
ComputationGraphConfiguration	conf()
ComputationGraph	init()
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, 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)

modelType

public Class<? extends Model> modelType()

conf

public ComputationGraphConfiguration conf()

init

public ComputationGraph init()

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)