這篇文章主要講解了如何使用PyQt的QLabel組件實現選定目標框功能,內容清晰明了,對此有興趣的小伙伴可以學習一下,相信大家閱讀完之后會有幫助。
問題背景
  基于PyQt5
開發了一個可以用于目標跟蹤的軟件,在開發過程中遇到一個問題,就是如何在PyQt5
的組件QLable
中自主選定目標框,這個在opencv
里面有專門的函數完成這個工作:cv2.selectROI()
,我的目的就是在QLabel
的基礎上,實現類似函數cv2.selectROI()
的功能,這樣在運行程序的過程中,就能在視頻框里面直接選取感興趣區域。
實現思路
具體要實現的功能是,在視頻顯示區域,點擊鼠標左鍵,開啟選擇,按照鼠標左鍵,移動游標,慢慢地繪制出紅色的目標框。釋放鼠標左鍵就停止選擇目標框。最開始以為PyQt
好歹也會提供這樣的類來進行開發吧,后來發現其實是沒有的,沒辦法只能寫一個QLabel
類的子類了。子類的命名為Label
,繼承自QLabel
類,在子類中重寫鼠標事件函數,接受鼠標在Label
對象上位置信號。PyQt
本來就有自己的事件循環,當鼠標落在視頻顯示區域的時候,觸發到Label
的鼠標事件,那么就可以開始繪制目標框了。
這里要記錄的就是鼠標按下左鍵時候的起始坐標pos_1
和移動坐標pos_2
,pos_1=(x0,y0)
,pos_2=(x1,y1)
。
重寫按下鼠標事件 按下鼠標左鍵,觸發事件函數mousePressEvent()
,事件函數打開繪制標志位self.select_roi_flag
,傳入事件對象數據,初始化起始坐標x0,y0
。
重寫釋放鼠標事件 按下鼠標左鍵,觸發事件函數mousePressEvent()
,關閉繪制標志位self.select_roi_flag
。
繪制事件 繼承鼠標事件繪制類,創建畫筆類對象,在這可以設置畫筆的顏色,畫線的粗細,如果繪制標志位self.select_roi_flag
是打開的,那么將事件對象的位置數據傳給x1,y1
。QRect
類是是PyQt
的內置數據結構,具體結構是這樣的Rect=(x,y,w,h)
,之后就調用畫筆對象方法動態繪制目標框。直到繪制標志位被關閉,就是釋放鼠標,則停止繪畫。
具體實現代碼:
from PyQt5.QtWidgets import QLabel from PyQt5.QtCore import Qt,QRect from PyQt5.QtGui import QPainter,QPen class Label(QLabel): x0=0 y0=0 x1=0 y1=0 open_mouse_flag=False select_roi_flag=False draw_roi_flag=False clear_flag=False rect = QRect() #按下鼠標 def mousePressEvent(self, event): if self.open_mouse_flag is True: self.select_roi_flag=True self.x0=event.x() self.y0=event.y() #釋放鼠標 def mouseReleaseEvent(self, event): self.select_roi_flag=False #移動鼠標 def mouseMoveEvent(self, event): if self.select_roi_flag is True: self.x1=event.x() self.y1=event.y() if self.draw_roi_flag is True: self.update() #繪制事件 def paintEvent(self,event): super().paintEvent(event) painter = QPainter(self) painter.setPen(QPen(Qt.red, 5, Qt.SolidLine)) if self.clear_flag is True: self.x0=0 self.y0=0 self.x1=0 self.y1=0 self.rect = QRect(self.x0, self.y0, abs(self.x1 - self.x0), abs(self.y1 - self.y0)) painter.drawRect(self.rect) self.update()
其他要注意的問題
子類Label
除了能自定義選擇目標框,還要在更新內容是清除繪制內容,實現這個功能可以通過設置清空標志位clear_flag
,當標志位打開的時候,將起始坐標和更新坐標重置為:(0,0)(0,0),這樣繪制內容就被更新了。
具體實現代碼:
# 清除label對象的繪制內容 def clear_label(self): self.label_show.clear_flag = True self.label_show.clear()
此外我還重寫了鍵盤事件,通過敲擊鍵盤來控制鼠標的繪制事件,這里的內容主要包括切換游標,開啟繪制事件,確認繪制事件。
具體實現代碼:
# 重寫鍵盤事件 def keyPressEvent(self, QKeyEvent): if self.open_keyboard_flag is True: # 當鍵盤事件為真的是才有鍵盤事件監控 if QKeyEvent.key() == Qt.Key_S: self.label_show.setCursor(Qt.CrossCursor) # 切換游標為十字型 self.label_show.open_mouse_flag = True self.label_show.draw_roi_flag = True if QKeyEvent.key() == Qt.Key_Q: # 按下'q'鍵鍵盤監控關閉 self.label_show.unsetCursor() self.label_show.draw_roi_flag = False self.label_show.open_mouse_flag = False self.open_keyboard_flag = False
看完上述內容,是不是對如何使用PyQt的QLabel組件實現選定目標框功能有進一步的了解,如果還想學習更多內容,歡迎關注億速云行業資訊頻道。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。