OpenCV--绘制各种图形

代码和笔记

import cv2
import numpy as np
from PIL import ImageFont, ImageDraw, Image


"""
绘制各种图形
"""

# 创建纯黑背景图
img = np.zeros((480, 640, 3), np.uint8)

""""
line：线，rectangle：矩形
pt1, pt2为图形的起始位置，thickness粗细，lineType控制光滑程度,shift一般不加
line(img, pt1, pt2, color, thickness, lineType, shift)，矩形(可用于捕捉人脸)类似
"""
# cv2.line(img, (10, 20), (300, 400), (0, 0, 255), 5, 4)
# cv2.line(img, (80, 100), (400, 500), (0, 0, 255), 5, 16)

"""
圆circle pt1(center), pt2(radius)这里变为圆心坐标和半径 注意OpenCV坐标行列和numpy中的相反
角度为逆时针
ellipse(img, center, axes, angel, startAngle, endAngle, color, thickness, lineType, shift)
椭圆 center圆心坐标, axes长短轴的一半  angel椭圆的角度 startAngle, endAngle从那个角度开始，到那个角度结束
"""

# cv2.circle(img, (320, 240), 50, (0, 0, 255), 5, 16)
# cv2.ellipse(img, (320, 240), (100, 50), 0, 0, 360, [0, 0, 255], 5, 16)

"""
绘制多边形polylines(img, pts, isClosed, color,thickness, lineType, shift )
pts 多边形点的坐标(一个列表, 三维(必须是int32位及以上的)), isClosed 是否闭合
填充多边形
"""

# pts = np.array([(250, 100), (150, 300), (50, 280)], np.int32)
# cv2.polylines(img, [pts], True, (0, 0, 255), 5)
# cv2.fillPoly(img, [pts], (0, 0, 255))

"""
绘制文本 putText(img, text, org, fontFace, fontScale, color, thickness, lineType, bottomLeftOrigin)
org文本左下角坐标 fontFace字体类型 fontScale大小
"""

cv2.putText(img, 'Hello OpenCv', (40, 40), cv2.FONT_HERSHEY_PLAIN, 5, [0, 0, 255])

# 纯白背景(也可以用图片)  text中文会乱码, 使用pillow包解决PIL
img1 = np.full((200, 200, 3), fill_value=255, dtype=np.uint8)
# 导入字体文件
font = ImageFont.truetype('./xx.ttc', 15)
# 创建一个pillow图片
img2 = Image.fromarray(img1)
draw = ImageDraw.Draw(img2)
# 利用draw去绘制中文
draw.text((10, 150), '你好呀', font=font, fill=(0, 255, 0, 0))
# 重新变回ndarray
img3 = np.array(img2)
cv2.imshow('img', img3)

cv2.imshow('draw', img)
cv2.waitKey(0)
cv2.destroyAllWindows()