SciPy Create 2D Polygon Mask

Question

I want to create a numpy 2D array which represents a binary mask of a polygon, using standard Python packages.

input: polygon vertices, image dimensions
output: binary mask of polygon (numpy 2D array)

 I also want to get the distance transform of this polygon using scipy.ndimage.morphology.distance_transform_edt.

Can anyone tell me how to accomplish this?

Answer 1

You can try the following code:

import numpy
from PIL import Image, ImageDraw

# polygon = [(x1,y1),(x2,y2),...] or [x1,y1,x2,y2,...]
# width = ?
# height = ?

img = Image.new('L', (width, height), 0)
ImageDraw.Draw(img).polygon(polygon, outline=1, fill=1)
mask = numpy.array(img)