nanobbox is a package that calculates Intersection over Union(IoU). This is based on cython_bbox and reimplemented using nanobind. Speedup is achieved by using nanobind and openmp.
pip install nanobbox
or
git clone https://github.com/neka-nat/nanobbox.git
cd nanobbox
pip install .
from nanobbox import bbox_overlaps
overlaps = bbox_overlaps(
np.ascontiguousarray(dt, dtype=np.float32),
np.ascontiguousarray(gt, dtype=np.float32)
)
time | |
---|---|
cython_bbox | 0.87s |
nanobbox | 0.77s |
nanobbox(omp) | 0.22s |