Here are some important pre-trained models offered by the DL4J Model Zoo API:
- VGG16: VGG-16 referred to in this paper: https://arxiv.org/abs/1409.1556.
This is a very deep convolutional neural network targeting large-scale image recognition tasks. We can use transfer learning to train the model further. All we have to do is import VGG16 from the model zoo:
ZooModel zooModel =VGG16.builder().build();
ComputationGraph network = (ComputationGraph)zooModel.initPretrained();
Note that the underlying architecture of the VGG16 model in the DL4J Model Zoo API is ComputationGraph.
- TinyYOLO: TinyYOLO is referred to in this paper: https://arxiv.org/pdf/1612.08242.pdf.
This is a real-time object detection model for fast and accurate image classification. We can apply transfer learning to this model as well after importing from it the model zoo, as shown here:
ComputationGraph pretrained = (ComputationGraph)TinyYOLO.builder().build().initPretrained();
Note that the underlying architecture of the TinyYOLO model in the DL4J model zoo API is ComputationGraph.
- Darknet19: Darknet19 is referred to in this paper: https://arxiv.org/pdf/1612.08242.pdf.
This is also known as YOLOV2, a faster object detection model for real-time object detection. We can apply transfer learning to this model after importing it from the model zoo, as shown here:
ComputationGraph pretrained = (ComputationGraph) Darknet19.builder().build().initPretrained();