ai.djl.modality.cv.zoo

Class YoloModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader

ai.djl.modality.cv.zoo.ObjectDetectionModelLoader

ai.djl.modality.cv.zoo.YoloModelLoader

All Implemented Interfaces:

ModelLoader

public class YoloModelLoader
extends ObjectDetectionModelLoader

A ModelLoader for YOLO models.

These models were built as part of the Gluon CV library and imported into DJL.

Yolo is a model to solve Application.CV.OBJECT_DETECTION. Prior models like ObjectDetectionModelLoader were built around classifiers and would classify at various locations in the image simultaneously. However, this is fairly inefficient. Yolo instead uses a regression that will predict both the bounding boxes and class probabilities leading to better performance and better precision, although it can increase localization errors (the boxes are less accurate). [paper]

YOLO is currently the best object detection model in the DJL Model Zoo in terms of both performance and prediction quality.

Field Summary

Fields inherited from class ai.djl.repository.zoo.BaseModelLoader

factories, modelZoo, resource

Constructor Summary

Constructors

Constructor and Description
YoloModelLoader(Repository repository, java.lang.String groupId, java.lang.String artifactId, java.lang.String version, ModelZoo modelZoo) Creates the Model loader from the given repository.

Method Summary

Methods inherited from class ai.djl.modality.cv.zoo.ObjectDetectionModelLoader

loadModel, loadModel, loadModel

Methods inherited from class ai.djl.repository.zoo.BaseModelLoader

createModel, getApplication, getArtifactId, listModels, loadModel, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

YoloModelLoader

public YoloModelLoader(Repository repository,
                       java.lang.String groupId,
                       java.lang.String artifactId,
                       java.lang.String version,
                       ModelZoo modelZoo)

Creates the Model loader from the given repository.

Parameters:

repository - the repository to load the model from

groupId - the group id of the model

artifactId - the artifact id of the model

version - the version number of the model

modelZoo - the modelZoo type that is being used to get supported engine types