問(wèn)題背景

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

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

實(shí)現(xiàn)思路

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

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

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

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

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

具體實(shí)現(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()

  #按下鼠標(biāo)
  def mousePressEvent(self, event):
    if self.open_mouse_flag is True:
      self.select_roi_flag=True
      self.x0=event.x()
      self.y0=event.y()

  #釋放鼠標(biāo)
  def mouseReleaseEvent(self, event):
    self.select_roi_flag=False

  #移動(dòng)鼠標(biāo)
  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()   

其他要注意的問(wèn)題

  子類Label除了能自定義選擇目標(biāo)框，還要在更新內(nèi)容是清除繪制內(nèi)容，實(shí)現(xiàn)這個(gè)功能可以通過(guò)設(shè)置清空標(biāo)志位clear_flag,當(dāng)標(biāo)志位打開的時(shí)候，將起始坐標(biāo)和更新坐標(biāo)重置為:(0，0)(0,0),這樣繪制內(nèi)容就被更新了。
具體實(shí)現(xiàn)代碼:

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

  此外我還重寫了鍵盤事件，通過(guò)敲擊鍵盤來(lái)控制鼠標(biāo)的繪制事件，這里的內(nèi)容主要包括切換游標(biāo)，開啟繪制事件，確認(rèn)繪制事件。

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

# 重寫鍵盤事件
def keyPressEvent(self, QKeyEvent):
  if self.open_keyboard_flag is True:         # 當(dāng)鍵盤事件為真的是才有鍵盤事件監(jiān)控
    if QKeyEvent.key() == Qt.Key_S:
      self.label_show.setCursor(Qt.CrossCursor)  # 切換游標(biāo)為十字型
      self.label_show.open_mouse_flag = True
      self.label_show.draw_roi_flag = True
    if QKeyEvent.key() == Qt.Key_Q:         # 按下'q'鍵鍵盤監(jiān)控關(guān)閉
      self.label_show.unsetCursor()
      self.label_show.draw_roi_flag = False
      self.label_show.open_mouse_flag = False
      self.open_keyboard_flag = False

到此這篇關(guān)于使用PyQt的QLabel組件實(shí)現(xiàn)選定目標(biāo)框功能的方法示例的文章就介紹到這了,更多相關(guān)PyQt QLabel選定目標(biāo)框 內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論