remove_small_boxes {torchvision}R Documentation

Remove Small Boxes

Description

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

Usage

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


[Package torchvision version 0.7.0 Index]