Crate powerboxesrs
source ·Expand description
Powerboxes is a package containing utility functions for transforming bounding boxes and computing metrics from them.
§Powerboxesrs
powerboxesrs
is a Rust package containing utility functions for transforming bounding boxes and computing metrics from them.
§Installation
Add the following to your Cargo.toml
file:
[dependencies]
powerboxesrs = "0.2.2"
§Usage
use ndarray::array;
use powerboxesrs::iou::iou_distance;
let boxes1 = array![[0.0, 0.0, 1.0, 1.0], [2.0, 2.0, 3.0, 3.0]];
let boxes2 = array![[0.5, 0.5, 1.5, 1.5], [2.5, 2.5, 3.5, 3.5]];
let iou = iou_distance(&boxes1, &boxes2);
assert_eq!(iou, array![[0.8571428571428572, 1.],[1., 0.8571428571428572]]);
§Functions available
warning: all functions expect the boxes to be in the format xyxy
(top left and bottom right corners) (not box conversion functions, of course)
§Box Transformations and utilities
box_areas
: Compute the area of list of boxesbox_convert
: Convert a box from one format to another. Supported formats arexyxy
,xywh
,cxcywh
.remove_small_boxes
: Remove boxes with area smaller than a thresholdmask_to_boxes
: Convert a mask to a list of boxes
§Box Metrics
iou_distance
: Compute the intersection over union matrix of two sets of boxesparallel_iou_distance
: Compute the intersection over union matrix of two sets of boxes in parallelgiou_distance
: Compute the generalized intersection over union matrix of two sets of boxesparallel_giou_distance
: Compute the generalized intersection over union matrix of two sets of boxes in parallel
§Box NMS
nms
: Non-maximum suppression, returns the indices of the boxes to keeprtree_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.