tfm.vision.box_ops.encode_boxes

Encodes boxes to targets.

tfm.vision.box_ops.encode_boxes(
    boxes, anchors, weights=None
)

Args
boxes	a tensor whose last dimension is 4 representing the coordinates of boxes in ymin, xmin, ymax, xmax order.
anchors	a tensor whose shape is the same as, or broadcastable to boxes, representing the coordinates of anchors in ymin, xmin, ymax, xmax order.
weights	None or a list of four float numbers used to scale coordinates.

Returns
encoded_boxes	a tensor whose shape is the same as boxes representing the encoded box targets.

Raises
ValueError	If the last dimension of boxes is not 4.