Skip to content

Speed up + New Features

Compare
Choose a tag to compare
@ZFTurbo ZFTurbo released this 25 Jun 17:58
· 46 commits to master since this release
  • Added many input data checks in all methods: set to 0.0 coords < 0, set to 1.0 coords > 1, swap cases where x2 < x1 and y2 < y1, remove zero area boxes.
  • Added numba @jit(nopython=True) for critical functions (NMS, NMW and WBF). Speed up around x2 times (tested on example_oid.py).
  • Added support for 3D boxes with function weighted_boxes_fusion_3d