《python學習筆記》 之 OpenCV庫 基礎功能測試②

• 時間：2020年2月29日  21時25分
• 對python環境下 OpenCV2 庫的基礎功能進行測試
• 功能包括：鼠標操作
• 測試代碼：

"""
功能：OpenCV2基礎測試
作者：劉澤文
時間：2020年2月29日  21時25分
"""
import cv2
import numpy as np

#鼠標按下時爲True,鬆開時爲False
drawing = False
#畫框框和畫線線切換
mode = True
#畫框框起始點點
ix,iy = -1,-1

#鼠標操作回調函數
def draw_circle(event,x,y,flags,param):
    global ix,iy,drawing,mode
    if event==cv2.EVENT_LBUTTONDBLCLK:#雙擊操作
        cv2.circle(img,(x,y),20,(255,0,0),-1)#畫圈圈
    elif event==cv2.EVENT_LBUTTONDOWN:#如果左鍵按下
        drawing=True#賦值起始點，開始畫畫
        ix,iy=x,y
    elif event==cv2.EVENT_MOUSEMOVE and flags==cv2.EVENT_FLAG_LBUTTON:#如果鼠標移動
        if drawing==True:#如果是畫畫模式
            if mode==True:#檢測模式，1爲畫框框，0爲畫圈圈
                cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)
            else:
                cv2.circle(img,(x,y),3,(0,0,255),-1)

    elif event==cv2.EVENT_LBUTTONUP:#如果松開左鍵
        drawing=False#停止畫圖

# 創建圖像與窗口並將窗口與回調函數綁定
img=np.zeros((512,512,3),np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_circle)#註冊鼠標回調函數

while(1):
    cv2.imshow('image',img)
    key = cv2.waitKey(1)
    if key==ord('m'):#'m'鍵切換模式
        mode = not mode
    elif key==27:#ESC退出
        break

cv2.destroyAllWindows()