python opencv3 轮廓检测
git:https://github.com/linyi0604/Computer-Vision
1 # coding:utf8
2
3 import cv2
4 import numpy as np
5
6 # 创建一个200*200 的黑色空白图像
7 img = np.zeros((200, 200), dtype=np.uint8)
8 # 在图像的中央位置 放置一个100*100的白色方块
9 img[50:150, 50: 150] = 255
10
11 cv2.imshow("image", img)
12 # 二值化操作
13 ret, thresh = cv2.threshold(img, 127, 255, 0)
14 """
15 ret, dst = cv2.threshold(src, thresh, value, type)
16 参数:
17 src: 原图像
18 thresh: 阈值
19 value: 新值 大于或小于阈值的值将赋新值
20 type: 方法类型,有如下取值:
21 cv2.THRESH_BINARY 黑白二值
22 cv2.THRESH_BINARY_INV 黑白二值翻转
23 cv2.THRESH_TRUNC 得到多像素值
24 cv2.THRESH_TOZERO
25 cv2.THRESH_TOZERO_INV
26 返回值:
27 ret: 得到的阈值值
28 dst: 阈值化后的图像
29 """
30
31 # 得到 修改后的图像, 轮廓, 轮廓的层次
32 image, contours, hierarchy = cv2.findContours(
33 thresh,
34 cv2.RETR_TREE,
35 cv2.CHAIN_APPROX_SIMPLE
36 )
37
38 """
39 img, contours, hierarchy = cv2.findContours(输入图像, 层次类型, 逼近方法)
40 参数:
41 输入图像: 该方法会修改输入图像,建议传入输入图像的拷贝
42 层次类型:
43 cv2.RETR_TREE 会得到图像中整体轮廓层次
44 cv2.RETR_EXTERNAL 只得到最外面的轮廓
45 逼近方法:
46
47 返回值:
48 img: 修改后的图像
49 contours: 图像的轮廓
50 hierarchy: 图像和轮廓的层次
51
52 """
53 # 原图像转换成bgr图像
54 color = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
55 # 用绿色 在原图像上画出轮廓
56 img = cv2.drawContours(color, contours, -1, (0, 255, 255), 2)
57
58 cv2.imshow("contours", color)
59 cv2.waitKey()
60 cv2.destroyAllWindows()