OpenCV的绘图函数，实力绘画篮球场

关键函数：cv2.line()，cv2.circle()，cv2.rectangle()，cv2.ellipse()，cv2.putText() 等。

绘制几何形状

import cv2 as cv
import numpy as np
'''
cv.rectangle()，cv.circle()，cv.line()，cv.putText()   分别是绘制矩形、圆、直线和文字。
cv.rectangle(img, (50, 50), (400, 400), (255, 255, 0), 4)
参数：图像对象，起始点，结束点，颜色，线粗细
cv.circle(img, (200, 200), 100, (255, 0, 0), -1, 8, 0)
参数：图像，圆心，半径，颜色，线粗细
cv.line(img, (100, 100), (400, 400), (0, 255, 0), 8)
参数：图像，起始点，结束点，颜色，线粗细
cv.putText(img, 'OpenCV', org, font, fontScale, color, thickness, cv.LINE_AA)
参数：图像，字符串，位置点，字体，字的大小，颜色，线粗细，cv.LINE_AA'''
img = np.zeros((800, 800, 3))temp = np.copy(img)# 绘制矩形
cv.rectangle(img, (100, 100), (400, 400), (255, 0, 0), 10)
# 绘制圆形
cv.circle(img, (250, 250), 100, (0, 0, 255), -1, 8, 0)
# 绘制直线
cv.line(img, (100, 100), (400, 400), (0, 255, 0), 8)
# 写文字
font = cv.FONT_HERSHEY_SIMPLEX
org = (0, 500)
fontScale = 4
color = (255, 255, 255) #白色  #color = (0, 0, 0)
thickness = 2
img = cv.putText(img, 'OpenCV', org, font, fontScale, color, thickness, cv.LINE_AA)cv.imshow('123', img)
cv.waitKey(0)

绘制OpenCV 的伪图标

import cv2 as cv
import numpy as np
img = np.ones((840, 840, 3), np.uint8) * 0    # img = np.ones((740, 600, 3), np.unit8)*255    unit8 --> uint8# 第一步 扇形 - 圆：
cv.ellipse(img, (420, 200), (140, 140), 120, 0, 300, (0, 0, 255), -1)
cv.circle(img, (420, 200), 55, (0, 0, 0), -1)  # cv.circle(img, (300, 140), 55, (25, 255, 255), -1)cv.ellipse(img, (260, 480), (140, 140), 0, 0, 300, (0, 255, 0), -1)
cv.circle(img, (260, 480), 55, (0, 0, 0), -1)cv.ellipse(img, (580, 480), (140, 140), 240, 0, 300, (255, 0, 0), -1)
cv.circle(img, (580, 480), 55, (0, 0, 0), -1)# 向图像添加文本
font = cv.FONT_HERSHEY_COMPLEX
# 图像，文字，位置，字体  字体大小 ，颜色  线条 ()
cv.putText(img, 'OpenCV', (100, 780), font, 5, (255, 255, 255), 10, cv.LINE_AA)cv.namedWindow('image', 0)
cv.imshow('image', img)
cv.waitKey(0)
cv.destroyAllWindows()'''plt.imshow(img)
plt.xticks([]), plt.yticks([])  # 隐藏 x 轴和 y 轴上的刻度值
plt.show()      # 按退出不起作用'''

效果：

绘制篮球场

先展示效果

代码

import cv2
import numpy as npdef DrawBorder(img, length, width, border, color, line_wide=2, line_type=cv2.LINE_AA):cv2.rectangle(img, (border, border), (border + length, border + width), color, line_wide, line_type)def DrawMidline(img, length, width, border, radius, color, line_wide=2, line_type=cv2.LINE_AA):cv2.line(img, (border + length // 2, border), (border + length // 2, border + width), color, line_wide, line_type)cv2.circle(img, (border + length // 2, border + width // 2), radius, color, line_wide, line_type)def DrawThreepointline(img, length, width, border, border_3pl, radius, color, line_wide=2, line_type=cv2.LINE_AA):'''border_3pl: 三分线半圆圆心与边线距离radius: 三分线半圆半径'''cv2.ellipse(img, (border + border_3pl, border + width // 2), (radius, radius), 0, -90, 90, color, line_wide,line_type)cv2.line(img, (border, border + width // 2 - radius), (border + border_3pl, border + width // 2 - radius), color,line_wide, line_type)cv2.line(img, (border, border + width // 2 + radius), (border + border_3pl, border + width // 2 + radius), color,line_wide, line_type)cv2.ellipse(img, (border + length - border_3pl, border + width // 2), (radius, radius), 180, -90, 90, color,line_wide, line_type)cv2.line(img, (border + length - border_3pl, border + width // 2 - radius),(border + length, border + width // 2 - radius), color, line_wide, line_type)cv2.line(img, (border + length - border_3pl, border + width // 2 + radius),(border + length, border + width // 2 + radius), color, line_wide, line_type)def Draw3szone(img, length, width, border, z_length, z_width, radius, color, zone_type=1, line_wide=2,line_type=cv2.LINE_AA):'''zone_type: 1矩形、2梯形'''if zone_type == 1:cv2.rectangle(img, (border, border + width // 2 - z_width // 2),(border + z_length, border + width // 2 + z_width // 2), color, line_wide, line_type)cv2.ellipse(img, (border + z_length, border + width // 2), (radius, radius), 0, -90, 90, color, line_wide,line_type)cv2.rectangle(img, (border + length - z_length, border + width // 2 - z_width // 2),(border + length, border + width // 2 + z_width // 2), color, line_wide, line_type)cv2.ellipse(img, (border + length - z_length, border + width // 2), (radius, radius), 180, -90, 90, color,line_wide, line_type)elif zone_type == 2:rect = np.array([[[border, border + width // 2 - z_width // 2],[border + z_length, border + width // 2 - radius],[border + z_length, border + width // 2 + radius],[border, border + width // 2 + z_width // 2]]], np.int32)cv2.polylines(img, rect, False, color, line_wide, line_type)cv2.circle(img, (border + z_length, border + width // 2), radius, color, line_wide, line_type)rect = np.array([[[border + length, border + width // 2 - z_width // 2],[border + length - z_length, border + width // 2 - radius],[border + length - z_length, border + width // 2 + radius],[border + length, border + width // 2 + z_width // 2]]], np.int32)cv2.polylines(img, rect, False, color, line_wide, line_type)cv2.circle(img, (border + length - z_length, border + width // 2), radius, color, line_wide, line_type)white = (255, 255, 255)
black = (0, 0, 0)
blue = (255, 0, 0)
green = (0, 255, 0)
red = (0, 0, 255)scale = 100
line_color = white
background_color = black# 球场各参数
border = int(2 * scale)  # 边界
length = int(28 * scale)  # 场地长
width = int(15 * scale)  # 场地宽
radius = int(1.8 * scale)  # 圆圈半径
# 三分线
radius_3pl = int(6.75 * scale)  # 三分线圆圈半径
border_3pl = int(1.57 * scale)  # 三分线圆心与边界距离
# 三秒区
type_3szone = 1  # 三秒区类型, 1为矩形, 2为梯形
length_3szone = int(5.8 * scale)  # 矩形/梯形长
width_3szone = int(4.9 * scale)  # 矩形/梯形宽img = (background_color * np.ones((width + border * 2, length + border * 2, 3))).astype(np.uint8)DrawBorder(img, length, width, border, line_color)
DrawMidline(img, length, width, border, radius, line_color)
DrawThreepointline(img, length, width, border, border_3pl, radius_3pl, line_color)
Draw3szone(img, length, width, border, length_3szone, width_3szone, radius, line_color, type_3szone)# 显示
cv2.namedWindow('test', 0)
cv2.resizeWindow('test', int(28 * 40 + 2 * 40), int(15 * 40 + 2 * 40))
cv2.imshow('test', img)
cv2.waitKey(0)
cv2.destroyAllWindows()# 存储
# cv2.imwrite("basketball_court_01.jpg", img)