#bounding-box #boxes #metrics

powerboxesrs

Utility functions to manipulate and compute metrics on boxes

9 releases

0.2.3 May 27, 2024
0.2.2 Feb 24, 2024
0.2.1 Jan 15, 2024
0.2.0 Dec 28, 2023
0.1.3 Nov 22, 2023

#865 in Algorithms

29 downloads per month

MIT license

82KB
1.5K SLoC

CI Coverage Crates.io

PowerBoxesrs

Powerboxes is a package containing utility functions for transforming bounding boxes and computing metrics.

Installation

cargo add powerboxesrs

Functions available

Box Transformations and utilities

  • box_areas: Compute the area of list of boxes
  • box_convert: Convert a box from one format to another. Supported formats are xyxy, xywh, cxcywh.
  • remove_small_boxes: Remove boxes with area smaller than a threshold
  • mask_to_boxes: Convert a mask to a list of boxes

Box Metrics

  • iou_distance: Compute the intersection over union matrix of two sets of boxes
  • parallel_iou_distance: Compute the intersection over union matrix of two sets of boxes in parallel
  • giou_distance: Compute the generalized intersection over union matrix of two sets of boxes
  • parallel_giou_distance: Compute the generalized intersection over union matrix of two sets of boxes in parallel
  • tiou_distance: Compute the tracking intersection over union matrix of two sets of boxes

Rotated Box Metrics

  • rotated_iou_distance: Compute the intersection over union matrix of two sets of rotated boxes in cxcywha format
  • rotated_giou_distance: Compute the generalized intersection over union matrix of two sets of rotated boxes in cxcywha format

Box NMS

  • nms: Non-maximum suppression, returns the indices of the boxes to keep
  • rtree_nms: Non-maximum suppression, returns the indices of the boxes to keep, uses a r-tree internally to avoid quadratic complexity, useful when having many boxes.

Use it in Rust

See the documentation for more details. Here is a simple example:

use ndarray::array;
use powerboxesrs::boxes::box_areas;
let boxes = array![[1., 2., 3., 4.], [0., 0., 10., 10.]];
let areas = box_areas(&boxes);
assert_eq!(areas, array![4., 100.]);

Dependencies

~4MB
~78K SLoC