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

Question

I have an image of a room and pixel coordinates from which I can generate lines. These lines separate the room into wall, floor and ceiling segments. I would like to create a segmentation map for this layout like in 1 . How can I do this in python?

Answer 1

Separate comment for visibility. Credit to @ZarakiKenpachi. I used cv2.fillPoly from open cv to generate polygons for each segment and combine them into one numpy array.