powerboxes 0.1.3

Utility functions to manipulate and compute metrics on boxes

PowerBoxes

Powerboxes is a package containing utility functions for transforming bounding boxes and computing metrics. It is implemented in both Python and Rust. It shows a significant speedup over the equivalent numpy implementations in Python, or other libraries such as shapely.

Installation

Python

pip install powerboxes

Rust

cargo add powerboxesrs

Python Usage

import powerboxes as pb
import numpy as np

# Create a bounding box
box = np.array([[0, 0, 1, 1]])

# Compute the area of the box
area = pb.box_areas(box)

# Compute the intersection of the box with itself
intersection = pb.iou_distance(box, box)

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

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

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![9., 121.]);