Python+OpenCV實現角度測量的示例代碼

Posted on by

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

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

event的賦值：

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

2、角度計算 

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

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)。以上函數便可根據三點的坐標值求其形成夾角的角度。

3、完整程序

# -*- coding: utf-8 -*-
'''
測量鼠標點擊過的三點形成的角度
'''
import cv2
import math
 
path = "picture_mqa\\angle_measure.bmp"  #圖片路徑
img = cv2.imread(path)
pointsList = []
 
#鼠標中斷觸發函數，將鼠標觸發事件位置處描點並將該點的坐標值紀錄入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]))
 
#根據相鄰的三點計算出其形成夾角的角度值
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) #鼠標觸發事件開啟
    
    if len(pointsList) % 3 ==0 and len(pointsList)!=0: #鼠標每觸發中斷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實現角度測量的示例代碼的文章就介紹到這瞭,更多相關Python OpenCV角度測量內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：