tvm.relay.vision

Vision network related operators.

Multibox operations.

tvm.relay.op.vision.multibox.multibox_prior(data, sizes=(1.0, ), ratios=(1.0, ), steps=(-1.0, -1.0), offsets=(0.5, 0.5), clip=False)

Generate prior(anchor) boxes from data, sizes and ratios.

Parameters:
  • data (relay.Expr) – The input data tensor.
  • sizes (tuple of float, optional) – Tuple of sizes for anchor boxes.
  • ratios (tuple of float, optional) – Tuple of ratios for anchor boxes.
  • steps (Tuple of float, optional) – Priorbox step across y and x, -1 for auto calculation.
  • offsets (tuple of int, optional) – Priorbox center offsets, y and x respectively.
  • clip (boolean, optional) – Whether to clip out-of-boundary boxes.
Returns:

out – 3-D tensor with shape [1, h_in * w_in * (num_sizes + num_ratios - 1), 4]

Return type:

relay.Expr

tvm.relay.op.vision.multibox.multibox_transform_loc(cls_prob, loc_pred, anchor, clip=True, threshold=0.01, variances=(0.1, 0.1, 0.2, 0.2))

Location transformation for multibox detection

Parameters:
  • cls_prob (tvm.relay.Expr) – Class probabilities.
  • loc_pred (tvm.relay.Expr) – Location regression predictions.
  • anchor (tvm.relay.Expr) – Prior anchor boxes.
  • clip (boolean, optional) – Whether to clip out-of-boundary boxes.
  • threshold (double, optional) – Threshold to be a positive prediction.
  • variances (Tuple of float, optional) – variances to be decoded from box regression output.
Returns:

ret

Return type:

tuple of tvm.relay.Expr

Non-maximum suppression operations.

tvm.relay.op.vision.nms.nms(data, valid_count, overlap_threshold=0.5, force_suppress=False, topk=-1)

Non-maximum suppression operator for object detection.

Parameters:
  • data (relay.Expr) – 3-D tensor with shape [batch_size, num_anchors, 6]. The last dimension should be in format of [class_id, score, box_left, box_top, box_right, box_bottom].
  • valid_count (relay.Expr) – 1-D tensor for valid number of boxes.
  • overlap_threshold (float, optional) – Non-maximum suppression threshold.
  • force_suppress (bool, optional) – Suppress all detections regardless of class_id.
  • topk (int, optional) – Keep maximum top k detections before nms, -1 for no limit.
Returns:

out – 3-D tensor with shape [batch_size, num_anchors, 6].

Return type:

relay.Expr