我想计算两个轴未对齐的矩形之间的交并比(IoU),这些矩形的轴角度小于30度。我也需要一个近似值。
一种可能的解决方案是检查两个矩形之间的角度是否小于30度,然后旋转它们使其轴平行对齐。从这里开始,计算IoU就变得很容易了。
另一种可能性是使用蒙特卡洛方法来计算交集(生成一个点,查找该点是否在一矩形的某条线下方和另一矩形的某条线上方),但这似乎代价高昂,因为我需要进行大量此类计算。
我希望有更好的方法;也许是几何库,或者是计算机视觉领域的算法。
我正在尝试使用深度神经网络学习抓取位置。我的算法需要预测RGB图像中对象的边界框(矩形)。对于任何图像,我还有真实的边界框(另一个矩形)。从这两个矩形中,我需要计算IoU。
有什么想法吗?
回答:
由于您在使用Python,我认为Shapely包可以满足您的需求。