Python根据坐标点生成多边形Mask的实现方法

江屿

发布时间：2026-10-14| 6人浏览过

原创

本文介绍如何在Python中根据给定的多边形顶点坐标生成二值化Mask。通过加载roi.mat文件获取x和y坐标，利用numpy将坐标合并并转换为int32格式，最后调用cv2.polylines和cv2.fillPoly函数绘制并填充多边形，生成仅包含0和1的Mask数组。

Python根据坐标点生成多边形Mask的实现方法

在处理数据集时，经常需要生成Mask，但源数据集中通常只提供了Mask顶点的坐标值。那么，如何在Python中根据这些坐标点生成仅由0和1表示的Mask区域呢？

主要借鉴cv2库中的相关方法来实现这一目标。

以我使用的数据情况为例，顶点坐标分别存储在roi.mat文件的x和y元素中。


matfn = 'roi.mat'
data = sio.loadmat(matfn)
x_cor = data['x']
y_cor = data['y']
im = np.zeros(图像对应尺寸, dtype="uint8")
cor_xy = np.hstack((x_cor, y_cor))
cv2.polylines(im, np.int32([cor_xy]), 1, 1)
cv2.fillPoly(im, np.int32([cor_xy]), 1)
mask_array = im

通过以上步骤即可得到所需的Mask。需要注意的是，坐标对最初是数组形式，必须先用方括号转换为列表形式，随后使用np.int32进行格式转换。这两步操作均包含在上述代码中，若省略任一环节，程序极易报错。

以上即为Python根据给定坐标点生成多边形Mask的完整实现过程，希望能为大家提供参考，也欢迎多多支持本站。