匀称多边形到二进制掩码

问题描述

我看到有人问过这个问题,但还没有真正找到完整的答复。我有一个简单的匀称多边形,名为 polygon。我想将此多边形提取为二进制掩码(理想情况下是一个 numpy 数组)。我该怎么做?

我还成功地从匀称转换为 geopandas,如图 here 所示,因此从 geopandas 中提取掩码也可以工作,但不幸的是,我还没有真正找到相关线索。

编辑: 需要明确的是,如果我要使用坐标网格,我的网格包含 x 和 y 笛卡尔坐标(无序),对应于构成形状轮廓的点。这些是浮点数,因此需要 int 输入的解决方案将无法正常工作。理想情况下,我希望起点是一个匀称的多边形而不是一组点,但是如果更可取,我可以使用一组无序的点(或者以某种方式从匀称的多边形中提取顺时针顶点)

我尝试过 Yusuke 描述的方法 here 但我得到的面具不太合理。

佑介的方法

#%% create grid and plot
nx,ny = 100,100
poly_verts = Plane1verts #this is a list of tuples containing cartesian coordinate pairs of the shape contour in x and y
# Create vertex coordinates for each grid cell...
# (<0,0> is at the top left of the grid in this system)
x,y = np.meshgrid(np.arange(nx),np.arange(ny))
x,y = x.flatten(),y.flatten()

points = np.vstack((x,y)).T

path = Path(poly_verts)
grid = path.contains_points(points)
grid = grid.reshape((ny,nx))

plt.imshow(grid)
plt.title('Grid plot')
plt.show()

掩码的结果图是

enter image description here

这不是我所期望的。而如下所述从 geopandas 绘图显示正确的形状。

#%% create shapely and plot for comparison
from shapely.geometry import polygon
#convert the sets of points dict to a shapely object
polygon1_plane1=polygon(Plane1vert_tuple)

p = gpd.GeoSeries(polygon1_plane1)
p.plot()
plt.show()

导致情节

enter image description here

编辑2: 这是我用作元组列表的坐标网格的副本

[(-8.982,-12.535),(-7.478,(-5.975,(-4.471,(-2.967,-11.031),(-1.463,(0.041,-9.527),(1.544,-8.023),(3.048,(4.552,(6.056,-6.52),(7.559,-5.016),(9.063,-3.512),(10.567,-2.008),-0.505),0.999),2.503),4.007),5.51),7.014),8.518),10.022),11.526),(-8.982,(-10.486,(-11.99,(-13.493,(-14.997,(-16.501,(-18.005,(-19.508,-12.535)]

解决方法

rasterio.features.rasterize 听起来正是您要找的。​​p>

from shapely.geometry import Polygon
import rasterio.features
import matplotlib.pyplot as plt

poly = Polygon([(0,50),(10,10),(30,0),(45,45),(0,50)])
img = rasterio.features.rasterize([poly],out_shape=(60,50))
plt.imshow(img)

rasterized image