Python - 将二进制掩码转换为多边形

Question

给定一个简单的二进制掩码(例如矩形的边界)。



如何使用多边形来获取 x-y 坐标？

这是我迄今为止尝试过的:

coords = np.transpose(np.nonzero(mask))

然而，这种方法会生成一个填充的对象，而不是所需的边界。

plt.plot(coords[:, 1], coords[:,0])

基本上，我想要一个白色像素的 x-y 坐标列表，以使用此列表重新绘制矩形(未填充)。

Answer 1

您可以使用 np.column_stack() + np.where() .这个想法是确定二值图像中的白色像素，然后在相应的 (x, y) 中排序。命令

coords = np.column_stack(np.where(image > 0))

另一种方法是使用 OpenCV 的 cv2.boundingRect() 找到边界矩形的坐标。 .这将为您提供宽度、高度和左上角 (x,y)坐标。这是一个查找坐标然后将多边形绘制到空白蒙版上的示例

import cv2
import numpy as np

image = cv2.imread('1.png', 0)
x,y,w,h = cv2.boundingRect(image)
mask = np.ones(image.shape, dtype=np.uint8) * 255
mask = cv2.merge([mask,mask,mask])
cv2.rectangle(mask, (x, y), (x + w, y + h), (36,255,12), 2)

cv2.imshow('mask', mask)
cv2.waitKey()