import cv2
import numpy as np
导入图片
image = cv2.imread("a.jpg")
转换为灰度图
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
用Sobel算子计算
x
,y
方向上的梯度,之后在x
方向上减去y
方向上的梯度,通过这个减法,我们留下具有高水平梯度和低垂直梯度的图像区域。
gradX = cv2.Sobel(gray, cv2.CV_32F, dx=1, dy=0, ksize=-1)
gradY = cv2.Sobel(gray, cv2.CV_32F, dx=0, dy=1, ksize=-1)
梯度减法
gradient = cv2.subtract(gradX, gradY)
gradient = cv2.convertScaleAbs(gradient)
去除图像上的噪声。
使用低通滤泼器平滑图像(9 x 9内核),这将有助于平滑图像中的高频噪声。
低通滤波器的目标是降低图像的变化率。如将每个像素替换为该像素周围像素的均值。
这样就可以平滑并替代那些强度变化明显的区域。
blurred = cv2.blur(gradient, (9, 9))
对模糊图像二值化,梯度图像中不大于90的任何像素都设置为0(黑色),否则,像素设置为255(白色)。
_, thresh = cv2.threshold(blurred, 90, 255, cv2.THRESH_BINARY)
用白色填充这些空余,使得后面的程序更容易识别,减少误差
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (25, 25))
closed = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel)
分别执行4次形态学腐蚀与膨胀
closed = cv2.erode(closed, None, iterations=4)
closed = cv2.dilate(closed, None, iterations=4)
找出域轮廓
cv2.findContours()
函数
第一个参数是要检索的图片,必须是为二值图,即黑白的(不是灰度图),
所以读取的图像要先转成灰度的,再转成二值图,我们在第三步用cv2.threshold()
函数已经得到了二值图。
第二个参数表示轮廓的检索模式,有四种:
cv2.RETR_EXTERNAL
表示只检测外轮廓
cv2.RETR_LIST
检测的轮廓不建立等级关系
cv2.RETR_CCOMP
建立两个等级的轮廓,上面的一层为外边界,里面的一层为内孔的边界信息。如果内孔内还有一个连通物体,这个物体的边界也在顶层。
cv2.RETR_TREE
建立一个等级树结构的轮廓。
第三个参数为轮廓的近似方法
cv2.CHAIN_APPROX_NONE
存储所有的轮廓点,相邻的两个点的像素位置差不超过1,即max(abs(x1-x2),abs(y2-y1))==1
cv2.CHAIN_APPROX_SIMPLE
压缩水平方向,垂直方向,对角线方向的元素,只保留该方向的终点坐标,例如一个矩形轮廓只需4个点来保存轮廓信息
cv2.findContours()
函数返回两个值,一个是轮廓本身,还有一个是每条轮廓对应的属性。
cv2.findContours()
函数返回第一个值是list,list中每个元素都是图像中的一个轮廓,用numpy
中的ndarray
表示。
每一个ndarray
里保存的是轮廓上的各个点的坐标。我们把list排序,点最多的那个轮廓就是我们要找的轮廓。
x = cv2.findContours(closed.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
_a, cnts, _b = x
c = sorted(cnts, key=cv2.contourArea, reverse=True)
# c 储存了所有的轮廓
cv2.minAreaRect()
函数:
主要求得包含点集最小面积的矩形,这个矩形是可以有偏转角度的,可以与图像的边界不平行。
rect = cv2.minAreaRect(c[0])
box = np.int0(cv2.boxPoints(rect))
OpenCV中通过
cv2.drawContours
在图像上绘制轮廓。
第一个参数是指明在哪幅图像上绘制轮廓
第二个参数是轮廓本身,在Python中是一个list
第三个参数指定绘制轮廓list
中的哪条轮廓,如果是-1
,则绘制其中的所有轮廓
第四个参数是轮廓线条的颜色
第五个参数是轮廓线条的粗细
cv2.drawContours(image, [box], -1, (0, 255, 0), 3)
裁剪保存图片。
box
里保存的是绿色矩形区域四个顶点的坐标。
找出四个顶点的x
,y
坐标的最大最小值。新图像的高=maxY-minY
,宽=maxX-minX
。
Xs = [i[0] for i in box]
Ys = [i[1] for i in box]
x1 = min(Xs)
x2 = max(Xs)
y1 = min(Ys)
y2 = max(Ys)
hight = y2 - y1
width = x2 - x1
cropImg = image[y1:y1+hight, x1:x1+width]
cv2.imwrite("b.jpg")
过程中,也可以通过来查看图片
cv2.namedWindow("img", 0)
cv2.imshow("img", image)
cv2.waitKey(0)