Clip boxes so that they lie inside an image of size size.

clip_boxes_to_image(boxes, size)

Arguments

boxes

(Tensor[N, 4]): boxes in \((x_{min}, y_{min}, x_{max}, y_{max})\) format with

  • \(0 \leq x_{min} < x_{max}\) and

  • \(0 \leq y_{min} < y_{max}\).

size

(Tuple[height, width]): size of the image

Value

clipped_boxes (Tensor[N, 4])