使用PyQt的QLabel組件實現(xiàn)選定目標框功能的方法示例

更新時間：2020年05月19日 09:06:21 作者：lonly_girl

這篇文章主要介紹了使用PyQt的QLabel組件實現(xiàn)選定目標框功能的方法示例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

問題背景

基于PyQt5開發(fā)了一個可以用于目標跟蹤的軟件，在開發(fā)過程中遇到一個問題，就是如何在PyQt5的組件QLable中自主選定目標框，這個在opencv里面有專門的函數(shù)完成這個工作:cv2.selectROI(),我的目的就是在QLabel的基礎上，實現(xiàn)類似函數(shù)cv2.selectROI()的功能，這樣在運行程序的過程中，就能在視頻框里面直接選取感興趣區(qū)域。直接貼出實現(xiàn)的最終效果:

上圖中的紅色框框就是在QLabel的基礎上實現(xiàn)的功能。

實現(xiàn)思路

具體要實現(xiàn)的功能是，在視頻顯示區(qū)域，點擊鼠標左鍵，開啟選擇，按照鼠標左鍵，移動游標，慢慢地繪制出紅色的目標框。釋放鼠標左鍵就停止選擇目標框。最開始以為PyQt好歹也會提供這樣的類來進行開發(fā)吧，后來發(fā)現(xiàn)其實是沒有的，沒辦法只能寫一個QLabel類的子類了。子類的命名為Label,繼承自QLabel類，在子類中重寫鼠標事件函數(shù)，接受鼠標在Label對象上位置信號。PyQt本來就有自己的事件循環(huán)，當鼠標落在視頻顯示區(qū)域的時候，觸發(fā)到Label的鼠標事件，那么就可以開始繪制目標框了。

這里要記錄的就是鼠標按下左鍵時候的起始坐標pos_1和移動坐標pos_2，pos_1=(x0,y0),pos_2=(x1,y1)。

重寫按下鼠標事件按下鼠標左鍵，觸發(fā)事件函數(shù)mousePressEvent(),事件函數(shù)打開繪制標志位self.select_roi_flag,傳入事件對象數(shù)據(jù)，初始化起始坐標x0，y0。

重寫釋放鼠標事件按下鼠標左鍵，觸發(fā)事件函數(shù)mousePressEvent(),關閉繪制標志位self.select_roi_flag。

繪制事件繼承鼠標事件繪制類，創(chuàng)建畫筆類對象，在這可以設置畫筆的顏色，畫線的粗細，如果繪制標志位self.select_roi_flag是打開的，那么將事件對象的位置數(shù)據(jù)傳給x1，y1。QRect類是是PyQt的內置數(shù)據(jù)結構，具體結構是這樣的Rect=(x,y,w,h)，之后就調用畫筆對象方法動態(tài)繪制目標框。直到繪制標志位被關閉，就是釋放鼠標，則停止繪畫。

具體實現(xiàn)代碼:

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除了能自定義選擇目標框，還要在更新內容是清除繪制內容，實現(xiàn)這個功能可以通過設置清空標志位clear_flag,當標志位打開的時候，將起始坐標和更新坐標重置為:(0，0)(0,0),這樣繪制內容就被更新了。
具體實現(xiàn)代碼:

# 清除label對象的繪制內容
def clear_label(self):
  self.label_show.clear_flag = True
  self.label_show.clear()

此外我還重寫了鍵盤事件，通過敲擊鍵盤來控制鼠標的繪制事件，這里的內容主要包括切換游標，開啟繪制事件，確認繪制事件。

具體實現(xiàn)代碼:

# 重寫鍵盤事件
def keyPressEvent(self, QKeyEvent):
  if self.open_keyboard_flag is True:         # 當鍵盤事件為真的是才有鍵盤事件監(jiān)控
    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'鍵鍵盤監(jiān)控關閉
      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組件實現(xiàn)選定目標框功能的方法示例的文章就介紹到這了,更多相關PyQt QLabel選定目標框內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: