Opencv - Contours 属性及操作 Python API

Opencv - Contours 属性及操作 Python API

Contours, 轮廓

• 定义 - Contours，explained simply as a curve joining all the continuous points (along the boundary), having same color or intensity.

• 用处 - shape analysis，object detection and recognition

  • 采用二值图(binary images)， 故，在寻找轮廓前，采用阈值或 canny 边缘检测.
  • cv2.findContours 函数在原图进行修改
  • Opencv中， cv2.findContours 类似于从黑色背景中寻找白色物体，因此，二值图中待寻找的物体应为白色，背景应为黑色

示例 - 寻找二值图的轮廓

import numpy as np
import cv2

im = cv2.imread('test.jpg')
imgray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(imgray,127,255,0)
image, contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

cv2.findContours函数输入有三个参数：
- thresh: source image
- cv2.RETR_TREE: 轮廓检索模式
- cv2.CHAIN_APPROX_SIMPLE: 轮廓逼近方法
输出三个结果：
- contours: 图像中所有的轮廓，python列表的形式保存. 每个单独的contour是包括物体边界点的(x,y)坐标的Numpy 数组.

示例 - 画出轮廓

img = cv2.drawContours(img, contours, -1, (0,255,0), 3) # 画出Image中的所有轮廓

img = cv2.drawContours(img, contours, 3, (0,255,0), 3) # 画出Image中的某个轮廓，比如第四个

# 一般也采用下面的方法画出某个轮廓，比如第四个
cnt = contours[4]
img = cv2.drawContours(img, [cnt], 0, (0,255,0), 3)

轮廓特征

主要有轮廓面积、周长、质心、边界框等.

1.Moments

Image Moments 用于计算物体质心、物体面积等.

import cv2
import numpy as np

img = cv2.imread('star.jpg',0)
ret,thresh = cv2.threshold(img,127,255,0)
contours,hierarchy = cv2.findContours(thresh, 1, 2)

cnt = contours[0]
M = cv2.moments(cnt) # 以dict形式输出所有moment值
print M

根据计算得到的 moments值，可以提取很多信息，比如面积、质心等.

2.Contour Area 轮廓面积

area = cv2.contourArea(cnt)

3.Contour Perimeter 轮廓周长

perimeter = cv2.arcLength(cnt,True)

4.Checking Convexity 轮廓凸性

k = cv2.isContourConvex(cnt)

5.Bounding Rectangle 边界框

5.1 Straight Bounding Rectangle 直矩形框

x,y,w,h = cv2.boundingRect(cnt) # (x,y)是rectangle的左上角坐标， (w,h)是width和height
img = cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0),2)

5.2 Rotated Rectangle 旋转矩形框

rect = cv2.minAreaRect(cnt) # 返回 Box2D结构，包括了左上角坐标(x,y),(width, height)和旋转角度
box = cv2.boxPoints(rect) # Rectangle的四个角
box = np.int0(box)
im = cv2.drawContours(im,[box],0,(0,0,255),2)

6.Minimum Enclosing Circle 最小包围圆

(x,y),radius = cv2.minEnclosingCircle(cnt)
center = (int(x),int(y))
radius = int(radius)
img = cv2.circle(img,center,radius,(0,255,0),2)

7.Fitting an Ellipse 椭圆拟合

ellipse = cv2.fitEllipse(cnt)
im = cv2.ellipse(im,ellipse,(0,255,0),2)

8.Fitting a Line 直线拟合

rows,cols = img.shape[:2]
[vx,vy,x,y] = cv2.fitLine(cnt, cv2.DIST_L2,0,0.01,0.01)
lefty = int((-x*vy/vx) + y)
righty = int(((cols-x)*vy/vx)+y)
img = cv2.line(img,(cols-1,righty),(0,lefty),(0,255,0),2)

9.Aspect Ratio 宽高比

物体边界框的宽高比.

x,y,w,h = cv2.boundingRect(cnt)
aspect_ratio = float(w)/h

10.Extent 轮廓物体面积与边界框面积比

area = cv2.contourArea(cnt)
x,y,w,h = cv2.boundingRect(cnt)
rect_area = w*h
extent = float(area)/rect_area

11.Solidity 轮廓物体面积和convex hull 面积比

area = cv2.contourArea(cnt)
hull = cv2.convexHull(cnt)
hull_area = cv2.contourArea(hull)
solidity = float(area)/hull_area

12.Equivalent Diameter 与轮廓物体面积相等的圆的直径

area = cv2.contourArea(cnt)
equi_diameter = np.sqrt(4*area/np.pi)

13.Mask and Pixel Points 物体的所有像素点

mask = np.zeros(imgray.shape,np.uint8)
cv2.drawContours(mask,[cnt],0,255,-1)
pixelpoints = np.transpose(np.nonzero(mask))
#pixelpoints = cv2.findNonZero(mask)

14.Maximum Value, Minimum Value and their locations

mask image 的最大值，最小值和其位置

min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(imgray,mask = mask)

15.Extreme Points 极值点

物体的最上、下、左、右的点.

leftmost = tuple(cnt[cnt[:,:,0].argmin()][0])
rightmost = tuple(cnt[cnt[:,:,0].argmax()][0])
topmost = tuple(cnt[cnt[:,:,1].argmin()][0])
bottommost = tuple(cnt[cnt[:,:,1].argmax()][0])

Reference

• Contours in OpenCV