如何为 python 中的图像创建分割图

How to create segmentation map for image in python

我有一张房间图像和像素坐标,我可以从中生成线条。这些线条将房间分成墙壁、地板和天花板部分。我想为 1 中的这种布局创建一个分割图。我如何在 python 中执行此操作?

单独评论可见性。感谢@ZarakiKenpachi。我使用 open cv 中的 cv2.fillPoly 为每个线段生成多边形并将它们组合成一个 numpy 数组。