Package | Description |
---|---|
org.deeplearning4j.zoo.model |
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 |
Darknet19
Darknet19
Reference: https://arxiv.org/pdf/1612.08242.pdf
|
class |
FaceNetNN4Small2
A variant of the original FaceNet model that relies on embeddings and triplet loss.
|
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 |
NASNet
U-Net
Implementation of NASNet-A in Deeplearning4j.
|
class |
ResNet50
Residual networks for deep learning.
|
class |
SimpleCNN
A simple convolutional network for generic image classification.
|
class |
SqueezeNet
U-Net
An implementation of SqueezeNet.
|
class |
TextGenerationLSTM
LSTM designed for text generation.
|
class |
TinyYOLO
Tiny YOLO
Reference: https://arxiv.org/pdf/1612.08242.pdf
|
class |
UNet
U-Net
An implementation of U-Net, a deep learning network for image segmentation in Deeplearning4j.
|
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)
|
class |
Xception
U-Net
An implementation of Xception in Deeplearning4j.
|
class |
YOLO2
YOLOv2
Reference: https://arxiv.org/pdf/1612.08242.pdf
|
Copyright © 2018. All rights reserved.