Uses of Class
org.deeplearning4j.nn.layers.objdetect.DetectedObject
-
Packages that use DetectedObject Package Description org.deeplearning4j.nn.adapters org.deeplearning4j.nn.layers.objdetect -
-
Uses of DetectedObject in org.deeplearning4j.nn.adapters
Methods in org.deeplearning4j.nn.adapters that return types with arguments of type DetectedObject Modifier and Type Method Description List<DetectedObject>
YoloModelAdapter. apply(Model model, INDArray[] inputs, INDArray[] masks, INDArray[] labelsMasks)
List<DetectedObject>
YoloModelAdapter. apply(INDArray... outputs)
-
Uses of DetectedObject in org.deeplearning4j.nn.layers.objdetect
Methods in org.deeplearning4j.nn.layers.objdetect that return types with arguments of type DetectedObject Modifier and Type Method Description List<DetectedObject>
Yolo2OutputLayer. getPredictedObjects(INDArray networkOutput, double threshold)
static List<DetectedObject>
YoloUtils. getPredictedObjects(INDArray boundingBoxPriors, INDArray networkOutput, double confThreshold, double nmsThreshold)
Given the network output and a detection threshold (in range 0 to 1) determine the objects detected by the network.
Supports minibatches - the returnedDetectedObject
instances have an example number index.
Note that the dimensions are grid cell units - for example, with 416x416 input, 32x downsampling by the network (before getting to the Yolo2OutputLayer) we have 13x13 grid cells (each corresponding to 32 pixels in the input image).Methods in org.deeplearning4j.nn.layers.objdetect with parameters of type DetectedObject Modifier and Type Method Description static double
YoloUtils. iou(DetectedObject o1, DetectedObject o2)
Returns intersection over union (IOU) between o1 and o2.Method parameters in org.deeplearning4j.nn.layers.objdetect with type arguments of type DetectedObject Modifier and Type Method Description static void
YoloUtils. nms(List<DetectedObject> objects, double iouThreshold)
Performs non-maximum suppression (NMS) on objects, using their IOU with threshold to match pairs.
-