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

Question

我有一个房间的图像和像素坐标，可以从中生成线条。这些线将房间分为墙，地板和天花板。我想像1一样为此布局创建一个细分地图。如何在python中执行此操作？

Answer 1

单独的注释以提高可见性。归功于@ZarakiKenpachi。我使用开放式cv中的cv2.fillPoly为每个线段生成多边形并将它们组合为一个numpy数组。