Remove boxes which contains at least one side smaller than min_size.

remove_small_boxes(boxes, min_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}\).

min_size

(float): minimum size

Value

keep (Tensor[K]): indices of the boxes that have both sides larger than min_size