圈子中的非最大抑制

时间:2018-12-20 06:02:31

标签: python machine-learning computer-vision object-detection non-maximum-suppression

我使用边界圆检测到对象。如何对它们执行非最大抑制?我既有矩形又有圆形,我想以某种方式将圆与圆分开,将矩形与矩形分开。我已经尝试对矩形使用非最大抑制,但是我不知道如何对圆形执行相同操作。

2 个答案:

答案 0 :(得分:2)

最近我不得不解决类似的问题。我所做的就是在圆的圆周上绘制一个假想的正方形,然后我基于面积重叠对正方形应用了非最大抑制。您可以在以下位置找到代码:https://github.com/Amanpradhan/Non-Max-Suppression

答案 1 :(得分:1)

我不认为您可以对圈子进行非最大抑制。至少,在非最大抑制之后,它们将不再是圆圈,更像ellipses。我不认为这是要走的路。

您可以做的是对矩形进行非最大抑制,找到正确的边界矩形后,您可以选择每个矩形边的中间点,并在通过所选4个点的矩形内部绘制圆/椭圆(您有一个椭圆的2个轴。