Learn R Programming

torchvision (version 0.7.0)

remove_small_boxes: Remove Small Boxes

Description

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

Usage

remove_small_boxes(boxes, min_size)

Value

keep (Tensor[K]): indices of the boxes that have both sides larger than 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