Shortcuts

remove_small_boxes

torchvision.ops.remove_small_boxes(boxes: torch.Tensor, min_size: float)torch.Tensor[source]

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

Parameters
  • boxes (Tensor[N, 4]) – boxes in (x1, y1, x2, y2) format with 0 <= x1 < x2 and 0 <= y1 < y2.

  • min_size (float) – minimum size

Returns

indices of the boxes that have both sides larger than min_size

Return type

Tensor[K]

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources