OpenCV:将单个圆拟合到图像(在 Python 中)

Jor*_*dan 5 python opencv computer-vision

我有这样的图像:

在此输入图像描述

我需要在黑暗区域中拟合一个椭圆(注意:必须是椭圆,而不是圆形)。在 OpenCV 中执行此操作的最佳方法是什么?到目前为止,我的第一步是对其应用自适应(Otsu)阈值,结果是:

在此输入图像描述

但我不知道从那里去哪里。我正在用 Python 编写应用程序,但它更多的是我正在寻找的算法设计。

根据回复/评论进行编辑:

好的,我已经尝试过形态学了。根据OpenCV 文档,我对其进行了 3 次迭代“关闭”操作(膨胀,然后腐蚀)以去除小颗粒,结果是:

在此输入图像描述

然后,为了将其扩展回更接近原始形状,我进行了 3 次迭代“打开”操作(腐蚀,然后膨胀),结果是:

在此输入图像描述

从这里开始,我进行了 Canny 边缘检测,结果是:

在此输入图像描述

现在,我使用findContours它,但遇到了问题。它沿着边缘发现了数十个轮廓,每个轮廓都是沿着圆周的一小段。这意味着,即使我采用最大尺寸的轮廓,它也可能只代表周长的 10%,这不足以精确地拟合椭圆。这就是为什么@Demi-Lune 建议的其他问题对我不起作用;它们都有非常干净、锐利的边缘,并findContours找到一个漂亮的单一轮廓,覆盖每个形状的整个边界,但对于我的混乱图像来说,这种情况不会发生。那么,从这里拟合椭圆的最佳方法是什么?

Ha *_*Bom 5

如果物体是圆形的,那么就cv2.minEnclosingCircle可以使用。或者,您可以使用它cv2.fitEllipse来查找对象周围最适合的椭圆。请记住在黑色背景中找到白色物体的轮廓。

import cv2
import numpy as np

img = cv2.imread("1.jpg")

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
_,thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
thresh = cv2.bitwise_not(thresh)

element = cv2.getStructuringElement(shape=cv2.MORPH_RECT, ksize=(5, 5))

morph_img = thresh.copy()
cv2.morphologyEx(src=thresh, op=cv2.MORPH_CLOSE, kernel=element, dst=morph_img)

contours,_ = cv2.findContours(morph_img,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

areas = [cv2.contourArea(c) for c in contours]
sorted_areas = np.sort(areas)

#bounding box (red)
cnt=contours[areas.index(sorted_areas[-1])] #the biggest contour
r = cv2.boundingRect(cnt)
cv2.rectangle(img,(r[0],r[1]),(r[0]+r[2],r[1]+r[3]),(0,0,255),2)

#min circle (green)
(x,y),radius = cv2.minEnclosingCircle(cnt)
center = (int(x),int(y))
radius = int(radius)
cv2.circle(img,center,radius,(0,255,0),2)

#fit ellipse (blue)
ellipse = cv2.fitEllipse(cnt)
cv2.ellipse(img,ellipse,(255,0,0),2)


cv2.imshow("morph_img",morph_img)
cv2.imshow("img", img)
cv2.waitKey()
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述 在此输入图像描述