python: 将多边形转为遮罩数组

Question

我有一个多边形，我想将其变成掩码数组，这样落在多边形 inside/outside 上的所有点都是 True/False。我以为我找到了完美的解决方案 (SciPy Create 2D Polygon Mask)，但由于某些原因，这不起作用！

我做错了什么？

#!/usr/bin/env python3

import numpy as np
import scipy as sp
from PIL import Image, ImageDraw

nx, ny = 10, 10

poly = np.array([(1, 1), (6, 2), (9, 9), (3, 7)])

img = Image.new("L", [nx, ny], 0)
ImageDraw.Draw(img).polygon(poly, outline=1, fill=1)
mask = np.array(img)

print(mask)
# [[1 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]
#  [0 0 0 0 0 0 0 0 0 0]]

更广泛的背景：我正在处理矩形网格上任意形状的特征。我有网格上所有边界点的索引，我想要这个特征周围的凸包的索引。 scipy.spatial.ConvexHull(boundary_points) 给了我凸包的边缘点，我现在想把这个包多边形变成蒙版。

Answer 1

poly 必须是元组列表或扁平化列表。出于某种原因，numpy 数组处理不当。您可以使用 poly.ravel().tolist() 或 list(map(tuple, poly)).

将多边形转换为 numpy 数组

python: 将多边形转为遮罩数组

python: turn polygon into mask array

python

numpy

scipy

point-in-polygon

python-imaging-library