Python-将二进制遮罩转换为多边形

Python - convert binary mask to polygon(Python-将二进制遮罩转换为多边形)

本文介绍了Python-将二进制遮罩转换为多边形的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

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

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

这是我到目前为止尝试过的内容:

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

但是,此方法生成的是填充对象,而不是所需的边界。

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

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

推荐答案

可以使用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()

这篇关于Python-将二进制遮罩转换为多边形的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:Python-将二进制遮罩转换为多边形

基础教程推荐