图像分割：创建多边形

Question

我输入的图像如下所示：

我喜欢以一种方式对图像进行分割，以获得仅包含水平垂直线的近似多边形。

我的第一种方法是霍夫分割，但我只能创建矩形对象。 这不适用于第二张图像。

然后我尝试使用决策树：对于每个图像，我训练了一个决策树，其中所有像素的输入x和y位置以及黑/白分类。 然后我只使用了这棵树的前n层。 使用这个新树，我对所有像素进行了预测。 有时这很好用，但有时它没有。 特别是树木深度因图片而异......

也许有人知道如何做到这一点？ 或者是否已有针对此用例的算法？

非常感谢你

问候

凯文

Answer 1

我得到了非常合理的结果，使用形态学“变薄”，然后是“侵蚀”，以消除水平或垂直方向的特征。 我只是在ImageMagick的命令行中执行它，但如果您愿意，可以使用Python绑定。

所以，横向特征：

convert poly.png -threshold 50% -morphology Thinning:-1 Skeleton -morphology erode rectangle:3x1 im1h.png

和垂直功能：

convert poly.png -threshold 50% -morphology Thinning:-1 Skeleton -morphology erode rectangle:1x3 im1v.png

并且，使用其他图像：

convert poly2.png -threshold 50% -morphology Thinning:-1 Skeleton -morphology erode rectangle:1x3 result.png