轮廓检测
cv2.findContours(image, mode, method)
-
mode:轮廓检索模式
- RETR_EXTERNAL: 只检索外面的轮廓
- RETR_LIST: 检索所有轮廓,并保存到一条链表中
- RETR_CCOMP: 检索所有轮廓,并将他们组织为两层,顶层是各部分的外部边界,第二层是空洞的边界
- RETR_TREE: 检索所有轮廓,并重构嵌套轮廓的整个层次
-
method:轮廓逼近方法
- CHAIN_APPROX_NONE: 以reeman链码的方式输出轮廓,所有其他方法输出多边形(顶点的序列)
- CHAIN_APPROX_SIMPLE: 压缩水平的,垂直的和斜的部分,也就是,函数只保留他们的终点部分
1. 读取图像
import cv2
import matplotlib.pyplot as plt
image = cv2.imread('6.png')
# 灰度图
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# 二值化
ret, thresh = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)
res = cv2.cvtColor(thresh, cv2.COLOR_BGR2RGB)
plt.imshow(res);
2. 寻找轮廓信息
contours, hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
3. 绘制轮廓
绘制所有轮廓
draw_img = image.copy()
# 传入绘制的底图, 绘制内容, 轮廓索引, 颜色, 厚度
res = cv2.drawContours(draw_img, contours, -![请添加图片描述](https://img-blog.csdnimg.cn/f3e31e527a67495eb65a49ce00a64f53.png)
1, (0, 255, 0), 2)
# BGR转RGB
res = cv2.cvtColor(res, cv2.COLOR_BGR2RGB)
plt.imshow(res);
绘制某一个索引(某一个图像的内轮廓或外轮廓)
draw_img = image.copy()
# 传入绘制的底图, 绘制内容, 轮廓索引, 颜色, 厚度
res = cv2.drawContours(draw_img, contours, 1, (0, 255, 0), 2)
# BGR转RGB
res = cv2.cvtColor(res, cv2.COLOR_BGR2RGB)
plt.imshow(res);
draw_img = image.copy()
# 传入绘制的底图, 绘制内容, 轮廓索引, 颜色, 厚度
res = cv2.drawContours(draw_img, contours, 0, (0, 255, 0), 2)
# BGR转RGB
res = cv2.cvtColor(res, cv2.COLOR_BGR2RGB)
plt.imshow(res);