使用open cv进行角度测量

使用open cv进行角度测量

用了一点初中数学的知识，准确度，跟鼠标点的准不准有关系，话不多说直接上代码

import cv2
import mathpath = "test.jpg"
img = cv2.imread(path)
pointsList = []def mousePoint(event, x, y, flags, params):# 如果鼠标左键按下if event == cv2.EVENT_LBUTTONDOWN:size = len(pointsList)if size != 0 and size % 3 != 0:cv2.line(img, tuple(pointsList[round((size - 1)/3) * 3]), (x, y), (0, 0, 255), 2)cv2.circle(img, (x, y), 5, (0, 0, 255), cv2.FILLED)pointsList.append([x, y])# 计算斜率
def gradient(pt1, pt2):return (pt2[1] - pt1[1])/(pt2[0] - pt1[0])def getAngle(pointsList):# 最后三个点pt1, pt2, pt3 = pointsList[-3:]# 计算两条直线的斜率m1 = gradient(pt1, pt2)m2 = gradient(pt1, pt3)angR = math.atan((m2 - m1)/(1 + (m2 * m1)))# 去掉小数,将弧度转换成角度angD = round(math.degrees(angR))cv2.putText(img, str(angD), ((pt1[0] - 40), pt1[1] - 20), cv2.FONT_HERSHEY_COMPLEX, 1.5, (0, 0, 255), 2)while True:if len(pointsList) % 3 == 0 and len(pointsList) != 0:getAngle(pointsList)cv2.imshow("Image", img)# 设置鼠标回调函数cv2.setMouseCallback("Image", mousePoint)# 是否在1毫秒内按下了'q'键if cv2.waitKey(1) & 0xFF == ord('q'):pointsList = []img = cv2.imread(path)

效果