【Python OpenCV】第三天：OpenCV 中的绘图函数

---

一、本期目标

自学 Python OpenCV 的第三天，今天我们来学习 OpenCV 中的绘图函数！

阅读本期文章，你将学会：

• 学习使用 OpenCV 绘制不同几何图形。

你将学习这些函数：

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

上面所有的这些绘图函数，都需要设置下面这些参数：

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

那我们就开始吧！

二、画线

要画一条线，你需要告诉函数这条线的起点和终点。

我们来画一条从左上方到右下角的蓝色线段。

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

三、画矩形

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

我们再来画一个绿色矩形，左上角坐标为（50,50），右下角坐标为（200,300）。

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线
cv2.rectangle(image, (50, 50), (200, 300), (0, 255, 0), 3) # 画矩形

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

四、画圆

画圆，只需要指定圆形的中心点坐标和半径大小。

我们来画一个红色的圆，半径为 60，中心点坐标为（400,400）.

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线
cv2.rectangle(image, (50, 50), (200, 300), (0, 255, 0), 3) # 画矩形
cv2.circle(image, (400,400), 60, (255, 0, 0), -1) # 画圆

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

五、画椭圆

画椭圆比较复杂，我们要多输入几个参数。

一个参数是中心点的位置坐标，一个参数是长轴和短轴的长度，还有椭圆沿逆时针方向旋转的角度。椭圆弧演顺时针方向起始的角度和结束角度，如果是 0 或 360，就是整个椭圆。

查看cv2.ellipse() 可以得到更多信息。下面的例子是在图片的中心（坐标为（256,256））绘制一个椭圆。

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线
cv2.rectangle(image, (50, 50), (200, 300), (0, 255, 0), 3) # 画矩形
cv2.circle(image, (400,400), 60, (255, 0, 0), -1) # 画圆
cv2.ellipse(image, (256, 256), (100, 50), 0, 0, 360, (255, 255, 255), 6) # 画椭圆

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

六、画多边形

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

这里画一个黄色的具有四个顶点的多边形。

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线
cv2.rectangle(image, (50, 50), (200, 300), (0, 255, 0), 3) # 画矩形
cv2.circle(image, (400,400), 60, (255, 0, 0), -1) # 画圆
cv2.ellipse(image, (256, 256), (100, 50), 0, 0, 360, (255, 255, 255), 6) # 画椭圆

# 画多边形
pts = numpy.array([[100, 50], [200, 300], [400, 200], [500, 100]], numpy.int32)
cv2.polylines(image, [pts], True, (255,255,0), 3)

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

注意：如果第三个参数是 False，我们得到的多边形是不闭合的（首尾不相连）。

注意：cv2.polylines() 可以被用来画很多条线。只需要把想要画的线放在一个列表中，将这个列表传给函数就可以了。每条线都会被独立绘制。这会比用cv2.line() 一条一条的绘制要快一些。

七、在图片上添加文字

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

• 你要绘制的文字
• 你要绘制的位置
• 字体类型（通过查看 cv2.putText() 的文档找到支持的字体）
• 字体的大小
• 文字的一般属性如颜色，粗细，线条的类型等。为了更好看一点推荐使用linetype=cv2.LINE_AA

在图像上绘制白色的 OpenCV：

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线
cv2.rectangle(image, (50, 50), (200, 300), (0, 255, 0), 3) # 画矩形
cv2.circle(image, (400,400), 60, (255, 0, 0), -1) # 画圆
cv2.ellipse(image, (256, 256), (100, 50), 0, 0, 360, (255, 255, 255), 6) # 画椭圆

# 画多边形
pts = numpy.array([[100, 50], [200, 300], [400, 200], [500, 100]], numpy.int32)
cv2.polylines(image, [pts], True, (255,255,0), 3)

# 写 “OpenCV”
font = cv2.FONT_HERSHEY_SIMPLEX # 字体
cv2.putText(image, 'OpenCV', (10, 500), font, 4, (255, 255, 255), 10)

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

八、完整代码及最终结果

下面就是完整代码及最终结果了，欢迎订阅专栏、关注作者。

代码：

import cv2
import numpy

image = numpy.zeros((512, 512, 3), numpy.uint8) # 创建图片
cv2.line(image, (0, 0), (511, 511), (0, 0, 255), 5) # 画线
cv2.rectangle(image, (50, 50), (200, 300), (0, 255, 0), 3) # 画矩形
cv2.circle(image, (400,400), 60, (255, 0, 0), -1) # 画圆
cv2.ellipse(image, (256, 256), (100, 50), 0, 0, 360, (255, 255, 255), 6) # 画椭圆

# 画多边形
pts = numpy.array([[100, 50], [200, 300], [400, 200], [500, 100]], numpy.int32)
cv2.polylines(image, [pts], True, (255,255,0), 3)

# 写 “OpenCV”
font = cv2.FONT_HERSHEY_SIMPLEX # 字体
cv2.putText(image, 'OpenCV', (10, 500), font, 4, (255, 255, 255), 10)

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) # 图像转化
cv2.imshow('Image', image)
cv2.waitKey(0)

结果：