我正在尝试使用机器学习(即随机森林)进行图像分割。分类器利用许多不同的像素级特征将像素分类为边缘像素或非边缘像素。我最近将我的分类器应用于一组甚至很难手动分割的图像(基于边缘像素图的图像分割),并且仍在努力从生成的概率图中获得合理的轮廓。我还将分类器应用于一组更简单的图像,当我将阈值调整为 0.95 时,我获得了相当好的预测轮廓(兰德指数 > 0.97)。我有兴趣通过过滤从概率图中提取的轮廓来改进分割结果。
这是原始图像:
专家概述:
从我的分类器生成的概率图:
当我基于 0.95 的阈值将图像转换为二进制时,这可以进一步细化:
我尝试在概率图中填充漏洞,但这给我留下了很多噪音,有时还会合并附近的单元格。我也尝试在 openCV 中寻找轮廓,但这也不起作用,因为这些轮廓中的许多轮廓都没有完全连接——轮廓中到处都会丢失一些像素。
编辑:我最终在概率图上使用了 Canny 边缘检测。