Python机器视觉--OpenCV入门--OpenCV鼠标绘制图形
1. 绘制图形
1.1 API介绍
利用OpenCV提供的绘制图形API可以轻松在图像上绘制各种图形, 比如直线, 矩形, 圆, 椭圆等图形.
-
line(img, pt1, pt2, color, thickness, lineType, shift) 画直线
- img: 在哪个图像上画线
- pt1, pt2: 开始点, 结束点. 指定线的开始与结束位置
- color: 颜色
- thickness: 线宽
- lineType: 线型.线型为-1, 4, 8, 16, 默认为8
- shift: 坐标缩放比例.
-
rectangle() 参数同上 画矩形
-
circle(img, center, radius, color[, thickness[, lineType[, shift]]]) 中括号内参数表示可选参数. 画圆
-
ellipse(img, 中心点, 长宽的一半, 角度, 从哪个角度开始, 从哪个角度结束,…)
- polylines(img, pts, isClosed, color[, thickness[, lineType[, shift]]]) 画多边形
- fillPoly 填充多边形
- putText(img, text, org, fontFace, fontScale, color[, thickness[, lineType[, bottomLeftOrigin]]]) 绘制文本
- text 要绘制的文本
- org 文本在图片中的左下角坐标
- fontFace 字体类型即字体
- fontScale 字体大小
1.2 代码实现
import cv2
import numpy as np
img = np.zeros((480, 640, 3), np.uint8)
# cv2.line(img, (10, 20), (300, 400), (0, 0, 255), 5, 4)
# cv2.line(img, (80, 100), (380, 480), (0, 0, 255), 5, 16)
# 画矩形
# cv2.rectangle(img, (10,10), (100, 100), (0, 0, 255), -1)
# 画圆
# cv2.circle(img, (320, 240), 100, (0, 0, 255))
# cv2.circle(img, (320, 240), 5, (0, 0, 255), -1)
# 画椭圆
# cv2.ellipse(img, (320, 240), (100, 50), 15, 0, 360, (0, 0, 255), -1)
#画多边形
# pts = np.array([(300, 10), (150, 100), (450, 100)], np.int32)
# cv2.polylines(img, [pts], True, (0, 0, 255))
#填充多边形
# cv2.fillPoly(img, [pts], (255, 255, 0))
cv2.putText(img, "Hello OpenCV!", (10, 400), cv2.FONT_HERSHEY_TRIPLEX, 3, (255,0,0))
cv2.imshow('draw', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
绘制中文 opencv本身不支持, 因为没有中文字体.我们可以借助pillow来实现绘制中文
# 安装pillow
import cv2
import numpy as np
from PIL import ImageFont, ImageDraw, Image
img = np.full((200, 200, 3), fill_value=255, dtype=np.uint8)
# 导入字体文件.
font_path = 'msyhbd.ttc'
font = ImageFont.truetype(font_path, 15)
img_pil = Image.fromarray(img)
draw = ImageDraw.Draw(img_pil)
draw.text((10, 150), '绘制中文', font=font, fill=(0, 255, 0, 0))
img = np.array(img_pil)
# 中文会显示问号
cv2.putText(img, '中文', (10, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 1)
cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
小惊喜:基于OpenCV实现按l键之后拖动鼠标绘制直线, 按r键之后拖动鼠标绘制矩形, 按r键拖动鼠标绘制圆形
# 按下l, 拖动鼠标, 可以绘制直线.
# 按下r, 拖到鼠标, 可以绘制矩形
# 按下c, 拖动鼠标, 可以绘制圆. 拖动的长度可以作为半径.
import cv2
import numpy as np
# 这是一个全局标志, 判断要画什么类型的图.
curshape = 0
startpos = (0, 0)
# 创建背景图
img = np.zeros((480, 640, 3), np.uint8)
# 要监听鼠标的行为, 所以必须通过鼠标回调函数实现.
def mouse_callback(event, x, y, flags, userdata):
# 引入全局变量
global curshape, startpos
# 引入非本层的局部变量用什么关键字nonlocal
if event == cv2.EVENT_LBUTTONDOWN:
# 记录起始位置
startpos = (x, y)
elif event ==0 and flags == 1: # 表示按下鼠标左键并移动鼠标
if curshape == 0: # 画直线
cv2.line(img, startpos, (x, y), (0, 0, 255), 1)
elif curshape == 1: # 画矩形
cv2.rectangle(img, startpos, (x, y), (0, 0, 255), 1)
elif curshape == 2: # 画圆
# 注意计算半径
a = (x - startpos[0])
b = (y - startpos[1])
r = int((a ** 2 + b ** 2) ** 0.5)
# 画圆的时候, 半径必须是整数
cv2.circle(img, startpos, r, (0, 0, 255), 1)
else: # 按其他的按键
print('暂不支持绘制其他图形')
elif event == cv2.EVENT_LBUTTONUP:
# 判断要画什么类型的图.
if curshape == 0: # 画直线
cv2.line(img, startpos, (x, y), (0, 0, 255), 3)
elif curshape == 1: # 画矩形
cv2.rectangle(img, startpos, (x, y), (0, 0, 255), 1)
elif curshape == 2: # 画圆
# 注意计算半径
a = (x - startpos[0])
b = (y - startpos[1])
r = int((a ** 2 + b ** 2) ** 0.5)
# 画圆的时候, 半径必须是整数
cv2.circle(img, startpos, r, (0, 0, 255), 1)
else: # 按其他的按键
print('暂不支持绘制其他图形')
# 创建窗口
cv2.namedWindow('drawshape', cv2.WINDOW_NORMAL)
# 设置鼠标回调函数
cv2.setMouseCallback('drawshape', mouse_callback)
while True:
cv2.imshow('drawshape', img)
# 检测按键
key = cv2.waitKey(1)
if key == ord('q'):
break
elif key == ord('l'):
curshape = 0
elif key == ord('r'):
curshape = 1
elif key == ord('c'):
curshape = 2
cv2.destroyAllWindows()
效果如图所示
