All Classes
-
Class Summary Class Description AlexNet AlexNet
contains a generic implementation of AlexNet adapted from [torchvision implmentation](https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py)AlexNet.Builder The Builder to construct aAlexNet
object.BasicModelZoo Model Zoo is a repository that contains all models for DJL.BasicZooProvider An imperative model zoo provider implements theZooProvider
interface.GoogLeNet GoogLeNet uses a stack of a total of 9 inception blocks and global average pooling to generate its estimates.GoogLeNet.Builder The Builder to construct aGoogLeNet
object.LeNet The model was introduced (and named for) Yann Lecun, for the purpose of recognizing handwritten digits in images [LeNet5](http://yann.lecun.com/exdb/lenet/).LeNet.Builder The Builder to construct aLeNet
object.Mlp Multilayer Perceptron (MLP) NeuralNetworks.MlpBlockFactory ABlockFactory
class that creates MLP block.MobileNetV1 MobileNetV1
contains a generic implementation of Mobilenet adapted from https://github.com/weiaicunzai/pytorch-cifar100/blob/master/models/mobilenet.py (Original author weiaicunzai).MobileNetV1.Builder The Builder to construct aMobileNetV1
object.MobileNetV2 MobileNetV2
contains a generic implementation of MobilenetV2 adapted from https://github.com/weiaicunzai/pytorch-cifar100/blob/master/models/mobilenetv2.py (Original author weiaicunzai).MobileNetV2.Builder The Builder to construct aMobileNetV2
object.NiN NiN uses convolutional layers with window shapes of 11×11 , 5×5 , and 3×3 , and the corresponding numbers of output channels are the same as in AlexNet.NiN.Builder The Builder to construct aNiN
object.ResnetBlockFactory ABlockFactory
class that createsResNetV1
block.ResNetV1 ResNetV1
contains a generic implementation of ResNet adapted from https://github.com/tornadomeet/ResNet/blob/master/symbol_resnet.py (Original author Wei Wu) by Antti-Pekka Hynninen.ResNetV1.Builder The Builder to construct aResNetV1
object.SimpleTextDecoder SimpleTextDecoder
implements aDecoder
that employs aRecurrentBlock
to decode text input.SimpleTextEncoder SimpleTextEncoder
implements aEncoder
that employs aRecurrentBlock
to encode text input.SingleShotDetection SingleShotDetection
is an implementation ofBlock
that implements a Single Shot Detection (SSD) model for object detection.SingleShotDetection.Builder The Builder to construct aSingleShotDetection
.SqueezeNet SqueezeNet
contains a generic implementation of Squeezenet adapted from [torchvision implmentation](https://github.com/pytorch/vision/blob/master/torchvision/models/squeezenet.py)SsdBlockFactory ABlockFactory
class that createsSingleShotDetection
block.TabNet TabNet
contains a generic implementation of TabNet adapted from https://towardsdatascience.com/implementing-tabnet-in-pytorch-fc977c383279 (Original author Samrat Thapa)TabNet.AttentionTransformer AttentionTransformer is where the tabNet models learn the relationship between relevant features, and decides which features to pass on to the feature transformer of the current decision step.TabNet.Builder The Builder to construct aTabNet
object.TabNet.DecisionStep DecisionStep is just combining featureTransformer and attentionTransformer together.VGG VGG model from the "Very Deep Convolutional Networks for Large-Scale Image Recognition" https://arxiv.org/abs/1409.1556 paper.VGG.Builder The Builder to construct aVGG
object.YOLOV3 YOLOV3
contains a generic implementation of yolov3 (Original author bubbliiiing).YOLOV3.Builder The Builder to construct aYOLOV3
object.