python - 计算轴未对齐的两个矩形之间的交集面积

Question

我想计算轴未对齐但轴角度小于 30 度的两个矩形之间的并集 IoU 的交集。还寻求一个近似值。

一种可能的解决方案是检查两个矩形之间的角度是否小于 30 度，然后平行旋转它们以对齐轴。从这里可以很容易地计算IoU。

另一种可能性是使用蒙特卡罗方法进行交集(生成一个点，查找该点是否位于一个矩形的某条线下方和另一个矩形的某条线上方)，但这似乎很昂贵，因为我需要使用此计算多次。

我希望有更好的东西；也许是一个几何库，或者可能是计算机视觉人员的算法。

我正在尝试使用深度神经网络学习抓取位置。我的算法应该预测 RGB 图像中对象的边界框(矩形)。对于任何图像，我还有地面实况(另一个矩形)边界框。我需要从这两个矩形中得到 IoU。

有什么想法吗？

Answer 1

由于您使用的是 Python，我认为 Shapely包将满足您的需求。