本文共 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/