org.deeplearning4j.nn.adapters

Class YoloModelAdapter

java.lang.Object

org.deeplearning4j.nn.adapters.YoloModelAdapter

All Implemented Interfaces:

Serializable, ModelAdapter<List<DetectedObject>>, OutputAdapter<List<DetectedObject>>

public class YoloModelAdapter
extends Object
implements ModelAdapter<List<DetectedObject>>

This ModelAdapter implementation is suited for use of Yolo2 model with ParallelInference

Author:

[email protected]

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
YoloModelAdapter()

Method Summary

Modifier and Type	Method and Description
List<DetectedObject>	apply(INDArray... outputs)
List<DetectedObject>	apply(Model model, INDArray[] inputs, INDArray[] masks, INDArray[] labelsMasks) This method invokes model internally, and does convertion to T

Methods inherited from class java.lang.Object

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

Constructor Detail

YoloModelAdapter

public YoloModelAdapter()

Method Detail

apply

public List<DetectedObject> apply(Model model,
                                  INDArray[] inputs,
                                  INDArray[] masks,
                                  INDArray[] labelsMasks)

Description copied from interface: ModelAdapter

This method invokes model internally, and does convertion to T

Specified by:

apply in interface ModelAdapter<List<DetectedObject>>

Returns:

apply

public List<DetectedObject> apply(INDArray... outputs)

Specified by:

apply in interface OutputAdapter<List<DetectedObject>>