Python+OpenCV 图像处理系列（3）—— 画线、矩形、画圆、画椭圆、画多边形

OpenCV 中的绘图函数

使用 OpenCV 绘制不同几何图形，其中包括的函数有

cv2.line()
cv2.circle()
cv2.rectangle()
cv2.ellipse()
cv2.putText()

以上函数均需要接受以下参数：

• img：图片的路径；
• color：形状的颜色。以 RGB 为例，需要传入一个元组，例如：（255,0,0）代表蓝色。对于灰度图只需要传入灰度值；
• thickness：线条的粗细。如果给一个闭合图形设置为 -1，那么这个图形就会被填充。默认值是 1；
• linetype：线条的类型，8 连接，抗锯齿等。默认情况是 8 连接。cv2.LINE_AA 为抗锯齿，这样看起来会非常平滑；

1. 画直线

要画一条线，你只需要告诉函数这条线的起点和终点。我们下面会画一条从左上方到右下角的蓝色线段。

import cv2

img = cv2.imread("2018-11-28_16-30-49.jpg")
# shape返回的是一个tuple元组，第一个元素表示图像的高度，第二个表示图像的宽度，第三个表示像素的通道数。
size = img.shape
print size  # (1728, 3072, 3)

cv2.line(img, pt1=(100, 0), pt2=(3072, 1000), color=(0, 255, 0), thickness=10)

# pt1=(x, y) x 表示当前位置与图片左边距的距离； y 表示当前位置与图片上边距的距离
# pt1, pt2 分别表示当前线的起始点和终点

cv2.imshow('aaa', img)
if cv2.waitKey(0) & 0xFF == ord('q'):
    cv2.destroyAllWindows()

图像矩阵的 shape 属性表示图像的大小，shape 会返回 tuple 元组，第一个元素表示矩阵行数，第二个元组表示矩阵列数，第三个元素是 3，表示像素值由光的三原色组成。

2. 画矩形

要画一个矩形，你需要告诉函数的左上角顶点和右下角顶点的坐标。

import cv2

img = cv2.imread("2018-11-28_16-30-49.jpg", cv2.IMREAD_COLOR)
size = img.shape
print size #  (1728, 3072, 3)

cv2.rectangle(img, pt1=(300, 0), pt2=(500, 150), color=(0, 255, 0), thickness=3)

cv2.imshow('aaa', img)
if cv2.waitKey(0) & 0xFF == ord('q'):
    cv2.destroyAllWindows()

3. 画圆

要画圆的话，只需要指定圆形的中心点坐标和半径大小。我们在上面的矩形中画一个圆。

import cv2

img = cv2.imread("2018-11-28_16-30-49.jpg", cv2.IMREAD_COLOR)
size = img.shape
print size # (1728, 3072, 3)

cv2.circle(img, center=(400, 60), radius=60, color=(0, 255, 255), thickness=-1)

cv2.imshow('aaa', img)
if cv2.waitKey(0) & 0xFF == ord('q'):
    cv2.destroyAllWindows()
    

4. 画椭圆

画椭圆比较复杂，我们要多输入几个参数。一个参数是中心点的位置坐标。下一个参数是长轴和短轴的长度。椭圆沿逆时针方向旋转的角度。椭圆弧演顺时针方向起始的角度和结束角度，如果是 0 和 360，就是整个椭圆。查看 cv2.ellipse() 可以得到更多信息。

import cv2

img = cv2.imread("2018-11-28_16-30-49.jpg", cv2.IMREAD_COLOR)

size = img.shape
print size # (1728, 3072, 3)

cv2.ellipse(img, center=(256, 256), axes=(100, 50), angle=0, startAngle=0, endAngle=180, color=(0, 255, 255), thickness=-1)

cv2.imshow('aaa', img)
if cv2.waitKey(0) & 0xFF == ord('q'):
    cv2.destroyAllWindows()
    

5. 画多边形

画多边形，需要指点每个顶点的坐标。用这些点的坐标构建一个大小等于行数 X1X2 的数组，行数就是点的数目。这个数组的数据类型必须为 int32。这里画一个黄色的具有四个顶点的多边形。

6. 在图片上添加文字

在图片上绘制文字，需要设置以下参数：

• 要绘制的文字 text
• 要绘制的位置 org
• 字体类型 fontFace
• 字体大小 fontScale
• 文字的一般属性，如颜色、粗细、线条的类型等。一般推荐使用 linetype=cv2.LINE_AA

import cv2

img = cv2.imread("2018-11-28_16-30-49.jpg", cv2.IMREAD_COLOR)

font = cv2.FONT_HERSHEY_SIMPLEX

cv2.putText(img, text='OpenCV', org=(10, 100), fontFace=font, fontScale=4, color=(0, 255, 255), thickness=4)
cv2.imshow('aaa', img)
if cv2.waitKey(0) & 0xFF == ord('q'):
    cv2.destroyAllWindows()