python 二维数组 轮廓提取
时间: 2024-01-09 21:23:06 浏览: 29
以下是Python实现二维数组轮廓提取的代码示例:
```python
import cv2
import numpy as np
def contour_extraction(img):
# 将彩色图像转换为灰度图像
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 二值化处理
ret, bin_img = cv2.threshold(gray_img, 127, 255, cv2.THRESH_BINARY)
# 轮廓提取
contours, hierarchy = cv2.findContours(bin_img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 绘制轮廓
contour_img = np.zeros_like(img)
cv2.drawContours(contour_img, contours, -1, (0, 255, 0), 2)
return contour_img
# 读入彩色图像
img_name = "./20210808/sample3.png"
img = cv2.imread(img_name)
# 调用轮廓提取函数
contour_img = contour_extraction(img)
# 显示结果
cv2.imshow("Contour Image", contour_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
这段代码首先将彩色图像转换为灰度图像,然后进行二值化处理。接下来使用`cv2.findContours()`函数提取轮廓,并使用`cv2.drawContours()`函数绘制轮廓。最后,显示结果图像。