How to connect broken lines in a binary image using Python/Opencv

二次信任 提交于 2019-12-30 00:38:50

问题


How can I make these lines connect at the target points? The image is a result of a skeletonization process.

I'm trying to segment each line as a region using Watershed Transform.


回答1:


MikeE's answer is quite good: using dilation and erosion morphological operations can help a lot in this context.
I want to suggest a little improvement, taking advantage of the specific structure of the image at hand. Instead of using dilation/erosion with a general kernel, I suggest using a horizontal kernel that will connect the endpoints of the horizontal lines, but will not connect adjacent lines to one another.

Here's a sketch of code (assuming the input image is stored in bw numpy 2D array):

import cv2, numpy as np

kernel = np.ones((1,20), np.uint8)  # note this is a horizontal kernel
d_im = cv2.dilate(bw, kernel, iterations=1)
e_im = cv2.erode(d_im, kernel, iterations=1) 

What you get is the dilated image:

Note how the gaps are closed, while maintaining the distinct horizontal lines

And the eroded image:

To remove artifacts created by dilate/erode, I suggest to extract the skeleton again.
If you further apply skeleton morphological operation to the eroded image you can get this result:

Once you have the curves connected you do not need to use watershed segmentation, but rather use connected components to label each curve.




回答2:


Since the image is already monochromatic, you can use morphological transformations to close broken lines.

In case you need an example, you can find it in the documentation here: http://docs.opencv.org/2.4/doc/tutorials/imgproc/opening_closing_hats/opening_closing_hats.html#closing

It works by first dilating the white areas in the image and then eroding back by the same amount. Effectively closing any holes in the white areas. More details and examples can be found here: http://docs.opencv.org/2.4/doc/tutorials/imgproc/erosion_dilatation/erosion_dilatation.html

This strategy requires that the gap in the broken line is smaller than the distance between neighboring lines.

It will not work, if the lines cross, or if the lines are too close to each other. However I think it will work quite nicely in your example.

You can also remove the artifacts below the third line using the erode function.



来源:https://stackoverflow.com/questions/43859750/how-to-connect-broken-lines-in-a-binary-image-using-python-opencv

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!