org.deeplearning4j.zoo.model

Class FaceNetNN4Small2

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.FaceNetNN4Small2

All Implemented Interfaces:

InstantiableModel

public class FaceNetNN4Small2
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
FaceNetNN4Small2(int numLabels, long seed)
FaceNetNN4Small2(int numLabels, long seed, WorkspaceMode workspaceMode)

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)
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

FaceNetNN4Small2

public FaceNetNN4Small2(int numLabels,
                        long seed)

FaceNetNN4Small2

public FaceNetNN4Small2(int numLabels,
                        long seed,
                        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()

conf

public ComputationGraphConfiguration conf()

init

public ComputationGraph init()

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)