ai.djl.mxnet.zoo.cv.objectdetection

Class YoloModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>

ai.djl.mxnet.zoo.cv.objectdetection.YoloModelLoader

All Implemented Interfaces:

ai.djl.repository.zoo.ModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>

public class YoloModelLoader
extends ai.djl.repository.zoo.BaseModelLoader<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>

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 SingleShotDetectionModelLoader 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, mrl, repository, version

Constructor Summary

Constructors

Constructor and Description
YoloModelLoader(ai.djl.repository.Repository repository) Constructs a YoloModelLoader given the repository, mrl, and version.

Method Summary

Modifier and Type	Method and Description
ai.djl.Application	getApplication()
ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>	loadModel(java.util.Map<java.lang.String,java.lang.String> filters, ai.djl.Device device, ai.djl.util.Progress progress)

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

createModel, getArtifactId, listModels, loadModel, match, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.repository.zoo.ModelLoader

loadModel, loadModel, loadModel, loadModel

Constructor Detail

YoloModelLoader

public YoloModelLoader(ai.djl.repository.Repository repository)

Constructs a YoloModelLoader given the repository, mrl, and version.

Parameters:

repository - the repository to load the model from

Method Detail

getApplication

public ai.djl.Application getApplication()

loadModel

public ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects> loadModel(java.util.Map<java.lang.String,java.lang.String> filters,
                                                                                                                    ai.djl.Device device,
                                                                                                                    ai.djl.util.Progress progress)
                                                                                                             throws java.io.IOException,
                                                                                                                    ai.djl.repository.zoo.ModelNotFoundException,
                                                                                                                    ai.djl.MalformedModelException

Throws:

java.io.IOException

ai.djl.repository.zoo.ModelNotFoundException

ai.djl.MalformedModelException