org.deeplearning4j.zoo.model

Class InceptionResNetV1

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.InceptionResNetV1

All Implemented Interfaces:

InstantiableModel

public class InceptionResNetV1
extends ZooModel

A variant of the original FaceNet model that relies on embeddings and triplet loss. Reference: https://arxiv.org/abs/1503.03832 Also based on the OpenFace implementation: http://reports-archive.adm.cs.cmu.edu/anon/2016/CMU-CS-16-118.pdf Revised and consolidated version by @crockpotveggies

Field Summary

Fields inherited from class org.deeplearning4j.zoo.ZooModel

ROOT_CACHE_DIR

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
ComputationGraphConfiguration.GraphBuilder	graphBuilder(String input)
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

InceptionResNetV1

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

InceptionResNetV1

public InceptionResNetV1(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(String input)

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)