Package 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>>
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description YoloModelAdapter()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description List<DetectedObject>
apply(Model model, INDArray[] inputs, INDArray[] masks, INDArray[] labelsMasks)
This method invokes model internally, and does convertion to TList<DetectedObject>
apply(INDArray... outputs)
-
-
-
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 interfaceModelAdapter<List<DetectedObject>>
- Returns:
-
apply
public List<DetectedObject> apply(INDArray... outputs)
- Specified by:
apply
in interfaceOutputAdapter<List<DetectedObject>>
-
-