mxnet.ndarray.contrib.MultiBoxDetection¶
-
mxnet.ndarray.contrib.
MultiBoxDetection
(cls_prob=None, loc_pred=None, anchor=None, clip=_Null, threshold=_Null, background_id=_Null, nms_threshold=_Null, force_suppress=_Null, variances=_Null, nms_topk=_Null, out=None, name=None, **kwargs)¶ Convert multibox detection predictions.
Parameters: - cls_prob (NDArray) – Class probabilities.
- loc_pred (NDArray) – Location regression predictions.
- anchor (NDArray) – Multibox prior anchor boxes
- clip (boolean, optional, default=1) – Clip out-of-boundary boxes.
- threshold (float, optional, default=0.01) – Threshold to be a positive prediction.
- background_id (int, optional, default='0') – Background id.
- nms_threshold (float, optional, default=0.5) – Non-maximum suppression threshold.
- force_suppress (boolean, optional, default=0) – Suppress all detections regardless of class_id.
- variances (tuple of <float>, optional, default=[0.1,0.1,0.2,0.2]) – Variances to be decoded from box regression output.
- nms_topk (int, optional, default='-1') – Keep maximum top k detections before nms, -1 for no limit.
- out (NDArray, optional) – The output NDArray to hold the result.
Returns: out – The output of this function.
Return type: NDArray or list of NDArrays