本文共 1658 字，大约阅读时间需要 5 分钟。

将二进制遮罩转换为多边形可以通过OpenCV库中的findContours()和approxPolyDP()函数来实现。以下是详细的实现步骤：

步骤1：导入必要的库确保安装了OpenCV库和numpy库。可以使用以下代码导入它们：

import cv2import numpy as np

步骤2：读取二进制遮罩图像使用cv2.imread()函数读取二进制遮罩图像。假设遮罩图像存储在binary_mask.png文件中，并将其读取为灰度图像：

mask = cv2.imread('binary_mask.png', 0)

步骤3：使用findContours()找到轮廓调用cv2.findContours()函数来找到图像中的轮廓。使用以下参数：

• mask：输入图像。
• cv2.RETR_EXTERNAL：只返回外部轮廓。
• cv2.CHAIN_APPROX_SIMPLE：简化轮廓，去除冗余点。

contours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

步骤4：绘制轮廓到原图上使用cv2.drawContours()函数绘制轮廓。设置轮廓颜色为红色，线条宽度为3：

result = cv2.drawContours(mask, contours, -1, (255, 0, 0), 3)cv2.imshow('Result', result)

步骤5：将轮廓转换为多边形如果需要将轮廓转换为多边形，可以使用cv2.approxPolyDP()函数。设置一个小的epsilon值，通常取弧长长度的百分比：

for contour in contours:    epsilon = 0.01 * cv2.arcLength(contour, True)    approx = cv2.approxPolyDP(contour, epsilon, True)    cv2.polylines(mask, [approx], True, (255, 0, 0), 3)

步骤6：显示转换后的多边形图像最后，显示转换后的多边形图像：

cv2.imshow('Polygon', mask)cv2.waitKey(0)cv2.destroyAllWindows()

完整代码示例

import cv2import numpy as np# 读取二进制遮罩图像mask = cv2.imread('binary_mask.png', 0)# 寻找轮廓contours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)# 绘制轮廓到原图上result = cv2.drawContours(mask, contours, -1, (255, 0, 0), 3)cv2.imshow('Result', result)# 如果需要转换为多边形for contour in contours:    epsilon = 0.01 * cv2.arcLength(contour, True)    approx = cv2.approxPolyDP(contour, epsilon, True)    cv2.polylines(mask, [approx], True, (255, 0, 0), 3)# 显示转换后的图像cv2.imshow('Polygon', mask)cv2.waitKey(0)cv2.destroyAllWindows()

测试用例

通过以上步骤，成功将二进制遮罩转换为多边形。

转载地址：http://ozlfk.baihongyu.com/