파이썬 SciPy Create 2D Polygon Mask

표준 Python 패키지를 사용하여 다각형의 이진 마스크를 나타내는 numpy 2D 배열을 만들어야합니다.

(더 큰 맥락 : scipy.ndimage.morphology.distance_transform_edt를 사용하여이 다각형의 거리 변환을 얻고 싶습니다.)

누구든지이 작업을 수행하는 방법을 보여줄 수 있습니까?

 

해결 방법

 

대답은 매우 간단합니다.

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)

 

참조 페이지 https://stackoverflow.com/questions/3654289