如何解决如何使用 OpenCV findContours 检测相交形状?
我在黑白图像中有两个相交的椭圆。我正在尝试使用 OpenCV findContours 使用此代码(以及下面的附图)将单独的形状识别为单独的轮廓。
import numpy as np
import matplotlib.pyplot as plt
import cv2
import skimage.morphology
img_3d = cv2.imread("C:/temp/test_annotation_overlap.png")
img_grey = cv2.cvtColor(img_3d,cv2.COLOR_BGR2GRAY)
contours = cv2.findContours(img_grey,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)[-2]
fig,ax = plt.subplots(len(contours)+1,1,figsize=(5,20))
thicker_img_grey = skimage.morphology.dilation(img_grey,skimage.morphology.disk(radius=3))
ax[0].set_title("ORIGINAL IMAGE")
ax[0].imshow(thicker_img_grey,cmap="Greys")
for i,contour in enumerate(contours):
new_img = np.zeros_like(img_grey)
cv2.drawContours(new_img,contour,-1,(255,255,255),10)
ax[i+1].set_title(f"Contour {i}")
ax[i+1].imshow(new_img,cmap="Greys")
plt.show()
然而找到了四个轮廓,没有一个是原始轮廓:
如何配置 OpenCV.findContours 来识别两个单独的形状? (注意我已经玩过霍夫圈,发现它对我正在分析的图像不可靠)
解决方法
从哲学上讲,您希望找到两个 圆,因为您在搜索它们时,需要的是中心和半径。从图形上看,这些数字是相连的,我们可以看到它们是分开的,因为我们知道什么是“圆”,并推断出坐标,与重叠的部分相匹配。
那么如何找到每个轮廓的最小封闭圆(或者在某些情况下 fitEllipse 并使用它们的参数):https://docs.opencv.org/master/dd/d49/tutorial_py_contour_features.html
然后说在清晰的图像中绘制该圆并获取不为零的像素坐标 - 通过掩码或通过逐步绘制圆来计算它们。
然后以某种精度将这些坐标与其他轮廓中的坐标进行比较,并将匹配的坐标附加到当前轮廓中。
最后:在清晰的画布上绘制扩展轮廓,并为单个非重叠圆应用 HoughCircles。 (或计算圆心和半径,圆的坐标并与轮廓进行精确比较。)
,也许我对这种方法有点矫枉过正,但它可以用作一种工作方法。您可以在图像上找到所有轮廓 - 您将获得两个类似“半圆”的轮廓,即相交的轮廓和作为两个相加圆的外部形状的轮廓。最小的三个轮廓应该是两个半圆和交点。如果您绘制这三个轮廓中的两个的组合,您将获得三个蒙版,其中两个将具有一个半圆和相交的组合。如果您对面具执行关闭操作,您将获得您的圆圈。然后你应该简单地做一个算法来检测哪两个面具代表一个完整的圆圈,你就会得到你的结果。这是示例解决方案:
import numpy as np
import cv2
# Function for returning solidity of contour - ratio of contour area to its
# convex hull area.
def checkSolidity(cnt):
area = cv2.contourArea(cnt)
hull = cv2.convexHull(cnt)
hull_area = cv2.contourArea(hull)
solidity = float(area)/hull_area
return solidity
img_orig = cv2.imread("circles.png")
# Had to dilate the image so the contour was completly connected.
img = cv2.dilate(img_orig,np.ones((3,3),np.uint8))
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) # Grayscale transformation.
# Otsu threshold.
thresh = cv2.threshold(gray,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)[1]
# Search for contours.
contours = cv2.findContours(thresh,cv2.CHAIN_APPROX_NONE,cv2.RETR_TREE)[0]
# Sorting contours from smallest to biggest.
contours.sort(key=lambda cnt: cv2.contourArea(cnt))
# Three contours - two semi circles and the intersection of the circles.
cnt1 = contours[0]
cnt2 = contours[1]
cnt3 = contours[2]
# Create three empty images
h,w = img.shape[:2]
mask1 = np.zeros((h,w),np.uint8)
mask2 = np.zeros((h,np.uint8)
mask3 = np.zeros((h,np.uint8)
# Draw all combinations of two out of three contours on the masks.
# The goal here is to draw one semicircle and the intersection together.
cv2.drawContours(mask1,[cnt1],(255,255),-1)
cv2.drawContours(mask1,[cnt3],-1)
cv2.drawContours(mask2,[cnt2],-1)
cv2.drawContours(mask2,-1)
cv2.drawContours(mask3,-1)
cv2.drawContours(mask3,-1)
# Perform closing operation on the masks so that you get uniform contours.
kernel_size = 25
kernel = np.ones((kernel_size,kernel_size),np.uint8)
mask1 = cv2.morphologyEx(mask1,cv2.MORPH_CLOSE,kernel)
mask2 = cv2.morphologyEx(mask2,kernel)
mask3 = cv2.morphologyEx(mask3,kernel)
masks = [] # List for storing all the masks.
masks.append(mask1)
masks.append(mask2)
masks.append(mask3)
# List where you will append solidity of the found biggest contour of every mask.
solidity = []
for mask in masks:
cnts = cv2.findContours(mask,cv2.RETR_TREE)[0]
cnt = max(cnts,key=lambda c: cv2.contourArea(c))
s = checkSolidity(cnt)
solidity.append(s)
# Index of the mask with smallest solidity.
min_solidity = solidity.index(min(solidity))
# The mask with the contour that has smallest solidity should be the one that
# has two semicirles drawn instead of one semicircle and the intersection.
#You could build a better function to check which mask is the one with
# two semicircles... like maybe the contour with the largest
# height and width of the bounding box etc.
# I chose solidity because it is enough for this example.
# Selection of colors.
colors = {
0: (0,1: (0,0),2: (255,}
# Draw contours of the mask other two masks - those two that have the
# semicircle and the intersection.
for i,s in enumerate(solidity):
if s != solidity[min_solidity]:
cnts = cv2.findContours(
masks[i],cv2.RETR_TREE)[0]
cnt = max(cnts,key=lambda c: cv2.contourArea(c))
cv2.drawContours(img_orig,[cnt],colors[i],1)
# Display result
cv2.imshow("img",img_orig)
cv2.waitKey(0)
cv2.destroyAllWindows()
结果:
,作为参考,我将根据这里的一些想法和其他一些想法发布我提出的解决方案。该解决方案的有效性为 99.9%,并且可以从包含许多重叠、包含和其他图像噪声(例如线条、文本等)的图像中恢复椭圆。
代码太长,散布在这里,但伪代码如下。
- 分割图像
- 使用 RETR_EXTERNAL 运行 cv2 findContours 以获取图像中的单独区域
- 对于每个图像,填充内部,应用蒙版,并独立于其他区域提取要处理的区域。
- 对每个区域独立执行剩余的步骤
- 使用 RETR_LIST 运行 cv2 findContours 以获取所有内部和外部轮廓
- 对于找到的每个轮廓,应用多边形平滑以减少像素化的影响
- 对于每个平滑的轮廓,确定该轮廓内具有相同曲率符号的连续线段,即完全向右弯曲或向左弯曲的线段(仅计算角度和符号变化)
- 在每个段内,使用最小二乘法拟合椭圆模型 (scikit-learn EllipseModel)
- 对原始图像执行 Lee 算法,计算每个像素与白色像素的最小距离
- 对于每个模型,执行贪婪的局部邻域搜索以改进对原始模型的拟合 - 拟合是拟合椭圆到白色像素的最大距离,来自 lee 算法的输出
不简单也不优雅,但对我处理的内容高度准确,通过对大量图像的人工审核得到确认。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。