Python+OpenCV實現(xiàn)角度測量的示例代碼

更新時間：2022年03月05日 11:35:27 作者：沒有水的海綿

本文介紹如何使用python語言實現(xiàn)角度測量，程序包括鼠標選點、直線斜率計算、角度計算三個子程序和一個主程序，感興趣的可以了解一下

本文介紹如何使用python語言實現(xiàn)角度測量，程序包括鼠標選點、直線斜率計算、角度計算三個子程序和一個主程序。最終實現(xiàn)效果：在圖片上用鼠標確認三點，程序將會顯示由此三點確定的角度，如下圖所示。

1、鼠標選點

# -*- coding: utf-8 -*-
 
import cv2
 
path = "picture_mqa\\angle_measure.bmp"
img = cv2.imread(path)
pointsList = []
 
def mousePoints(event,x,y,flags,params):
    if event  == cv2.EVENT_LBUTTONDOWN:
        cv2.circle(img,(x,y),5,(0,0,255),cv2.FILLED)
        print(x,y)
 
while True:
    cv2.imshow('Image', img)
    cv2.setMouseCallback('Image',mousePoints)
    
    key_scan = cv2.waitKey(1) & 0xff 
    
    if key_scan == ord("q"):
        pointsList = []
        img = cv2.imread(path)
    elif key_scan == ord("s"):
        break
    
cv2.destroyAllWindows()

while循環(huán)內cv2.setMouseCallback('Image',mousePoints)為鼠標中斷觸發(fā)事件的開啟函數(shù)，作用是當在Image圖片上鼠標觸發(fā)中斷事件時，程序跳轉到mousePoints()中斷服務函數(shù)內，并給mousePoints()的五個入口參數(shù)event,x,y,flags,params賦值。其中， event是cv2_EVENT_* (MouseEventTypes)類型的變量，為鼠標觸發(fā)中斷事件的類型；x和y為鼠標觸發(fā)中斷事件時在image圖像的橫縱坐標；flags是cv2_EVENT_FLAG_* (MouseEventFlags)類型的變量，為特殊中斷事件的標志位；param是用戶自定義的參數(shù)。本文的程序中使用 EVENT_LBUTTONDOW#左鍵點擊觸發(fā)事件，當鼠標左鍵點擊時，標注該點并記錄其坐標。

event的賦值：

EVENT_MOUSEMOVE #滑動
EVENT_LBUTTONDOWN #左鍵點擊
EVENT_RBUTTONDOWN #右鍵點擊
EVENT_MBUTTONDOWN #中鍵點擊
EVENT_LBUTTONUP #左鍵放開
EVENT_RBUTTONUP #右鍵放開
EVENT_MBUTTONUP #中鍵放開
EVENT_LBUTTONDBLCLK #左鍵雙擊
EVENT_RBUTTONDBLCLK #右鍵雙擊
EVENT_MBUTTONDBLCLK #中鍵雙擊

2、角度計算

由1可以得到鼠標點擊位置處的坐標，我們將其放入pointList列表內。當列表內的坐標數(shù)目為3的倍數(shù)時調用getAngle()函數(shù)，計算出三點確定的兩條直線的夾角。

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 = abs(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))

由直線的兩點式方程可得直線的傾斜角為angle = arctan(y2-y1,x2-x1)，則兩條直線的夾角為angle0 =angle1-angle2 = arctan(y2-y1,x2-x1) - arctan(y2-y3,x2-x3)。以上函數(shù)便可根據(jù)三點的坐標值求其形成夾角的角度。

3、完整程序

# -*- coding: utf-8 -*-
'''
測量鼠標點擊過的三點形成的角度
'''
import cv2
import math
 
path = "picture_mqa\\angle_measure.bmp"  #圖片路徑
img = cv2.imread(path)
pointsList = []
 
#鼠標中斷觸發(fā)函數(shù)，將鼠標觸發(fā)事件位置處描點并將該點的坐標值紀錄入pointList列表內
#連接相鄰三點使其形成一個夾角
def mousePoints(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))
        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]))
 
#根據(jù)相鄰的三點計算出其形成夾角的角度值
def getAngle(pointsList):
    pt1,pt2,pt3 = pointsList[-3:] 
    m1 = gradient(pt1, pt2)
    m2 = gradient(pt1, pt3)
    angR = abs(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))       
 
 
while True:
    cv2.imshow('Image', img) #圖片顯示
    cv2.setMouseCallback('Image',mousePoints) #鼠標觸發(fā)事件開啟
    
    if len(pointsList) % 3 ==0 and len(pointsList)!=0: #鼠標每觸發(fā)中斷3次計算一次其形式夾角的角度值
        getAngle(pointsList)
    
    key_scan = cv2.waitKey(1) & 0xff #鍵盤掃描
    if key_scan == ord("q"):  #輸入'q'時圖片刷新
        pointsList = []
        img = cv2.imread(path)
        
    elif key_scan == ord("s"): #輸入's'時退出程序
        break
    
cv2.destroyAllWindows()

到此這篇關于Python+OpenCV實現(xiàn)角度測量的示例代碼的文章就介紹到這了,更多相關Python OpenCV角度測量內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: