这篇文章主要讲解了如何使用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进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。