Package | Description |
---|---|
org.deeplearning4j.zoo | |
org.deeplearning4j.zoo.model |
Modifier and Type | Method and Description |
---|---|
static Map<ZooType,ZooModel> |
ModelSelector.select(int numLabels,
int seed,
int iterations,
WorkspaceMode workspaceMode,
ZooType... zooTypes)
Select specific models from the zoo.
|
static Map<ZooType,ZooModel> |
ModelSelector.select(WorkspaceMode workspaceMode,
ZooType... zooTypes) |
static Map<ZooType,ZooModel> |
ModelSelector.select(ZooType... zooTypes) |
static Map<ZooType,ZooModel> |
ModelSelector.select(ZooType zooType) |
static Map<ZooType,ZooModel> |
ModelSelector.select(ZooType zooType,
int numLabels) |
static Map<ZooType,ZooModel> |
ModelSelector.select(ZooType zooType,
int numLabels,
int seed,
int iterations) |
static Map<ZooType,ZooModel> |
ModelSelector.select(ZooType zooType,
int numLabels,
int seed,
int iterations,
WorkspaceMode workspaceMode)
Select multiple models from the zoo according to type.
|
static Map<ZooType,ZooModel> |
ModelSelector.select(ZooType zooType,
int numLabels,
WorkspaceMode workspaceMode) |
Modifier and Type | Class and Description |
---|---|
class |
AlexNet
AlexNet
Dl4j's AlexNet model interpretation based on the original paper ImageNet Classification with Deep Convolutional Neural Networks
and the imagenetExample code referenced.
|
class |
FaceNetNN4Small2
A variant of the original FaceNet model that relies on embeddings and triplet loss.
|
class |
GoogLeNet
GoogleLeNet
Reference: http://arxiv.org/pdf/1409.4842v1.pdf
Revised and consolidated version by @kedardoshi
|
class |
InceptionResNetV1
A variant of the original FaceNet model that relies on embeddings and triplet loss.
|
class |
LeNet
LeNet was an early promising achiever on the ImageNet dataset.
|
class |
ResNet50
Residual networks for deep learning.
|
class |
SimpleCNN
A simple convolutional network for generic image classification.
|
class |
TextGenerationLSTM
LSTM designed for text generation.
|
class |
VGG16
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
|
class |
VGG19
VGG-19, from Very Deep Convolutional Networks for Large-Scale Image Recognition
https://arxiv.org/abs/1409.1556)
|
Copyright © 2017. All rights reserved.