opencv学习笔记六——轮廓检测

轮廓检测

cv2.findContours(image, mode, method)

1. mode:轮廓检索模式

   • RETR_EXTERNAL: 只检索外面的轮廓
   • RETR_LIST: 检索所有轮廓，并保存到一条链表中
   • RETR_CCOMP: 检索所有轮廓，并将他们组织为两层，顶层是各部分的外部边界，第二层是空洞的边界
   • RETR_TREE: 检索所有轮廓，并重构嵌套轮廓的整个层次

2. 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);