从选择性搜索算法的opencv实现中获取区域掩码

Question

我正在尝试生成用于训练分割模型的区域建议。我看到了 OpenCV 的实现：

cv2.ximgproc.segmentation.createSelectiveSearchSegmentation()

这个实现很好地返回了区域提案作为边界框。但根据我对这个算法的理解，它使用 Felzenszwalb 的图分割作为初始种子，并运行颜色、大小、纹理和形状相似性度量来进一步对区域进行分组。

有没有办法访问在 OpenCV 中对 Felzenszwalb aglorithm 的输出运行选择性搜索后生成的分段掩码？

Answer 1

如果您希望可视化选择性搜索的分段掩码输出，我已向 OpenCV-contrib （未致力于原始存储库）文件添加了功能，现在与框一起，您还可以将分段掩码作为输出。

在代码中，图像是私有变量，但掩模在 OpenCV 中存储为点。因此，为了检索图像蒙版，我对代码中返回的项目进行了一些更新。