Shortcuts

remove_small_boxes

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

Remove every box from boxes which contains at least one side length that is smaller than min_size.

Note

For sanitizing a BoundingBoxes object, consider using the transform SanitizeBoundingBoxes() instead.

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