如何检测圆形腐蚀/膨胀

Question

我想检测一条线上的圆形腐蚀和膨胀。对于膨胀，我尝试递归腐蚀图像，并且在每次递归时，我检查宽度/高度纵横比。如果比率小于 4，我假设它的轮廓是圆形的，对于每个这样的轮廓，我从力矩和面积计算圆心和半径。这是检测循环膨胀的函数：

def detect_circular_dilations(img, contours):
    contours_current, hierarchy = cv2.findContours(img, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    if len(contours_current) == 0:
        return get_circles_from_contours(contours)
    for c in contours_current:
        x, y, w, h = cv2.boundingRect(c)
        if w > h:
            aspect_ratio = float(w) / h
        else:
            aspect_ratio = float(h) / w
        if aspect_ratio < 4 and w < 20 and h < 20 and w > 5 and h > 5:
            contours.append(c)
    return detect_circular_dilations(cv2.erode(img, None, iterations=1), contours)

我想检测的循环膨胀示例如下：

我还没有解决的另一个问题是圆腐蚀的检测。循环腐蚀的一个例子如下：

在这里，我用红色矩形标记了我想检测的圆形侵蚀。可能有一些较小的圆形图案（左侧）不应被视为实际的圆形侵蚀。

有谁知道检测这种圆形的最佳方法是什么？对于循环膨胀，我将不胜感激任何评论/建议，以便有可能使检测更加稳健。

谢谢！

Answer 1

我会尝试用cv2.Canny() 找到线的两条边并搜索轮廓。如果您按边界框的宽度对轮廓进行排序，则前两个轮廓将是您的线条边缘。之后，您可以计算一个边缘中每个点到另一边缘的最小距离。然后您可以计算距离的中位数，并说如果一个点的距离大于或小于中位数（+- 容差），那么该点就是线的膨胀或腐蚀，并将其附加到列表中。如果需要，您可以通过遍历列表来整理噪音，如果它们不连续（在 x 轴上），则删除它们。

这是一个简单的例子：

import cv2
import numpy as np
from scipy import spatial

def detect_dilation(median, mindist, tolerance):
    count = 0
    for i in mindist:
        if i > median + tolerance:
            dilate.append((reshape_e1[count][0], reshape_e1[count][1]))
        elif i < median - tolerance:
            erode.append((reshape_e1[count][0], reshape_e1[count][1]))
        else:
            pass
        count+=1

def other_axis(dilate, cnt):
    temp = []
    for i in dilate:
        temp.append(i[0])
    for i in cnt:
        if i[0] in temp:
            dilate.append((i[0],i[1]))

img = cv2.imread('1.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
edges = cv2.Canny(gray,100,200)
_, contours, hierarchy = cv2.findContours(edges,cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)
contours.sort(key= lambda cnt :cv2.boundingRect(cnt)[3])
edge_1 = contours[0]
edge_2 = contours[1]
reshape_e1 = np.reshape(edge_1, (-1,2))
reshape_e2 =np.reshape(edge_2, (-1,2))
tree = spatial.cKDTree(reshape_e2)
mindist, minid = tree.query(reshape_e1)
median = np.median(mindist)
dilate = []
erode = []
detect_dilation(median,mindist,5)
other_axis(dilate, reshape_e2)
other_axis(erode, reshape_e2)

dilate = np.array(dilate).reshape((-1,1,2)).astype(np.int32)
erode = np.array(erode).reshape((-1,1,2)).astype(np.int32)
x,y,w,h = cv2.boundingRect(dilate)
cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
x,y,w,h = cv2.boundingRect(erode)
cv2.rectangle(img,(x,y),(x+w,y+h),(0,0,255),2)

cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

结果：

编辑：

如果图片有一条断线（这意味着更多的轮廓），您必须将每个轮廓视为单独的线。您可以通过在cv2.boundingRect() 的帮助下创建感兴趣的区域来实现此目的。但是当我尝试使用新上传的图片时，该过程不是很稳健，因为您必须更改容差才能获得所需的结果。由于我不知道其他图像是什么样的，您可能需要一种更好的方法来获取平均距离和容差因子。这里的任何方式都是我所描述的示例（15 表示容差）：

import cv2
import numpy as np
from scipy import spatial

def detect_dilation(median, mindist, tolerance):
    count = 0
    for i in mindist:
        if i > median + tolerance:
            dilate.append((reshape_e1[count][0], reshape_e1[count][1]))
        elif i < median - tolerance:
            erode.append((reshape_e1[count][0], reshape_e1[count][1]))
        else:
            pass
        count+=1

def other_axis(dilate, cnt):
    temp = []
    for i in dilate:
        temp.append(i[0])
    for i in cnt:
        if i[0] in temp:
            dilate.append((i[0],i[1]))

img = cv2.imread('2.jpg')
gray_original = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
_, thresh_original = cv2.threshold(gray_original, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)

# Filling holes
_, contours, hierarchy = cv2.findContours(thresh_original,cv2.RETR_CCOMP,cv2.CHAIN_APPROX_SIMPLE)
for cnt in contours:
    cv2.drawContours(thresh_original,[cnt],0,255,-1)
_, contours, hierarchy = cv2.findContours(thresh_original,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)

for cnt in contours:
    x2,y,w2,h = cv2.boundingRect(cnt)
    thresh = thresh_original[0:img.shape[:2][1], x2+20:x2+w2-20] # Region of interest for every "line"
    edges = cv2.Canny(thresh,100,200)
    _, contours, hierarchy = cv2.findContours(edges,cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)
    contours.sort(key= lambda cnt: cv2.boundingRect(cnt)[3])
    edge_1 = contours[0]
    edge_2 = contours[1]
    reshape_e1 = np.reshape(edge_1, (-1,2))
    reshape_e2 =np.reshape(edge_2, (-1,2))
    tree = spatial.cKDTree(reshape_e2)
    mindist, minid = tree.query(reshape_e1)
    median = np.median(mindist)
    dilate = []
    erode = []
    detect_dilation(median,mindist,15)
    other_axis(dilate, reshape_e2)
    other_axis(erode, reshape_e2)
    dilate = np.array(dilate).reshape((-1,1,2)).astype(np.int32)
    erode = np.array(erode).reshape((-1,1,2)).astype(np.int32)
    x,y,w,h = cv2.boundingRect(dilate)
    if len(dilate) > 0:
        cv2.rectangle(img[0:img.shape[:2][1], x2+20:x2+w2-20],(x,y),(x+w,y+h),(255,0,0),2)
    x,y,w,h = cv2.boundingRect(erode)
    if len(erode) > 0:
        cv2.rectangle(img[0:img.shape[:2][1], x2+20:x2+w2-20],(x,y),(x+w,y+h),(0,0,255),2)

cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

结果：

Answer 2

此类问题通常使用距离变换和中轴变换来解决。这些在某种程度上是相关的，因为中轴沿着距离变换的脊延伸。总体思路是：

1. 计算图像的距离变换（对于每个前景像素，返回到最近的背景像素的距离；有些库以另一种方式实现，在这种情况下，您需要计算倒置图像的距离变换)。

2. 计算中轴（或骨架）。

3. 沿中轴的距离变换值是相关值，我们忽略所有其他像素。在这里，我们看到了线的局部半径。

4. 局部最大值是膨胀的质心。使用阈值来确定哪些是重要的膨胀，哪些不是（嘈杂的轮廓会导致许多局部最大值）。

5. 局部最小值是侵蚀的质心。

例如，我使用下面的 MATLAB 代码得到了以下输出。

这是我使用的代码。它使用带有DIPimage 3 的MATLAB，作为原理的快速证明。使用您喜欢使用的任何图像处理库，这应该可以直接转换为 Python。

% Read in image and remove the red markup:
img = readim('https://i.stack.imgur.com/bNOTn.jpg');
img = img{3}>100;
img = closing(img,5);

% This is the algorithm described above:
img = fillholes(img);               % Get rid of holes
radius = dt(img);                   % Distance transform
m = bskeleton(img);                 % Medial axis
radius(~m) = 0;                     % Ignore all pixels outside the medial axis
detection = dilation(radius,25)==radius & radius>25; % Local maxima with radius > 25
pos = findcoord(detection);         % Coordinates of detections
radius = double(radius(detection)); % Radii of detections

% This is just to make the markup:
detection = newim(img,'bin');
for ii=1:numel(radius)
   detection = drawshape(detection,2*radius(ii),pos(ii,:),'disk');
end
overlay(img,detection)