mxnet.npx.multibox_detection

multibox_detection(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.00999999978) – 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