PowerBoxesrs

Powerboxes 是一个包含转换边界框和计算度量的实用函数的包。

安装

cargo add powerboxesrs

可用函数

盒子转换和实用工具

• box_areas: 计算盒子列表的面积
• box_convert: 将一个盒子从一个格式转换为另一个格式。支持的格式有 xyxy，xywh，cxcywh。
• remove_small_boxes: 删除面积小于阈值的盒子
• mask_to_boxes: 将掩码转换为盒子列表

盒子度量

• iou_distance: 计算两组盒子的交并比矩阵
• parallel_iou_distance: 并行计算两组盒子的交并比矩阵
• giou_distance: 计算两组盒子的广义交并比矩阵
• parallel_giou_distance: 并行计算两组盒子的广义交并比矩阵
• tiou_distance: 计算两组盒子的跟踪交并比矩阵

旋转盒子度量

• rotated_iou_distance: 计算以 cxcywha 格式表示的两个旋转盒子集合的交并比矩阵
• rotated_giou_distance: 计算以 cxcywha 格式表示的两个旋转盒子集合的广义交并比矩阵

盒子 NMS

• nms: 非最大值抑制，返回要保留的盒子的索引
• rtree_nms: 非最大值抑制，返回要保留的盒子的索引，内部使用 r-tree 以避免二次复杂度，当有大量盒子时很有用。

在 Rust 中使用它

查看 文档 了解更多详情。以下是一个简单的示例

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.]);