QLabel 用鼠标画选框 并显示框在图片上 选框全部保留除非右键清除

<think>我们面临的任务是在QLabel上显示图片，并允许用户通过鼠标绘制多个矩形选框（ROI），同时保留所有选框直到用户右键点击清除。以下是实现思路： 1. **继承QLabel**：创建一个自定义的QLabel类，重写鼠标事件（mousePressEvent, mouseMoveEvent, mouseReleaseEvent）来实现选框的绘制。 2. **存储选框**：使用一个列表来存储所有已经绘制好的矩形（QRect对象）。 3. **当前选框**：在鼠标拖动过程中，实时绘制当前正在绘制的矩形，但只有在释放鼠标时才将其添加到存储列表中。 4. **绘制事件**：重写paintEvent，在QLabel上绘制图片，并绘制所有存储的矩形以及当前正在绘制的矩形（如果有）。 5. **右键清除**：在鼠标事件中检测右键点击，清除所有存储的矩形。 ### 步骤详解 #### 1. 创建自定义QLabel类 ```python from PyQt5.QtWidgets import QLabel, QApplication from PyQt5.QtCore import Qt, QPoint, QRect from PyQt5.QtGui import QPixmap, QPainter, QPen, QColor class ImageLabel(QLabel): def __init__(self, parent=None): super().__init__(parent) self.setMouseTracking(False) self.setAlignment(Qt.AlignCenter) self.origin = QPoint() # 鼠标按下时的起始点 self.current_rect = None # 当前正在绘制的矩形（尚未完成） self.rectangles = [] # 存储所有已完成的矩形 def setPixmap(self, pixmap): super().setPixmap(pixmap) self.pix = pixmap # 保存原始图片 # 其他事件处理... ``` #### 2. 鼠标事件处理 - **mousePressEvent**: - 左键按下：记录起始点，并开始绘制新矩形。 - 右键按下：清除所有矩形。 - **mouseMoveEvent**: - 左键拖动：更新当前矩形的终点，并重绘（实时显示矩形）。 - **mouseReleaseEvent**: - 左键释放：完成当前矩形，将其添加到矩形列表中，重置当前矩形。 ```python def mousePressEvent(self, event): if event.button() == Qt.LeftButton: self.origin = event.pos() self.current_rect = QRect(self.origin, self.origin) self.update() # 触发重绘 elif event.button() == Qt.RightButton: # 清除所有矩形 self.rectangles.clear() self.current_rect = None self.update() super().mousePressEvent(event) def mouseMoveEvent(self, event): if self.current_rect is not None and event.buttons() & Qt.LeftButton: # 根据当前鼠标位置更新矩形 self.current_rect = QRect(self.origin, event.pos()).normalized() self.update() super().mouseMoveEvent(event) def mouseReleaseEvent(self, event): if event.button() == Qt.LeftButton and self.current_rect is not None: # 完成当前矩形 self.rectangles.append(self.current_rect) self.current_rect = None self.update() super().mouseReleaseEvent(event) ``` #### 3. 重写paintEvent 在绘制事件中，先绘制图片，然后绘制所有已存储的矩形，最后绘制当前正在绘制的矩形（如果有）。 ```python def paintEvent(self, event): super().paintEvent(event) # 先调用父类绘制（显示图片） # 创建绘图工具 painter = QPainter(self) pen = QPen(QColor(255, 0, 0), 2) # 红色，2像素宽 painter.setPen(pen) # 绘制所有已完成的矩形 for rect in self.rectangles: painter.drawRect(rect) # 绘制当前正在绘制的矩形（如果有） if self.current_rect is not None: painter.drawRect(self.current_rect) ``` #### 4. 使用自定义QLabel 在主窗口中使用这个自定义的QLabel来显示图片。 ```python class MainWindow(QWidget): def __init__(self): super().__init__() layout = QVBoxLayout() self.label = ImageLabel() layout.addWidget(self.label) self.setLayout(layout) # 加载图片 pixmap = QPixmap("image.jpg") # 替换为你的图片路径 self.label.setPixmap(pixmap) ``` ### 注意事项 1. **坐标系统**：QLabel中绘制的矩形坐标是相对于QLabel自身的，因此如果图片缩放或居中等情况，需要调整坐标。上述代码中，由于我们调用了`super().paintEvent(event)`，它会绘制QLabel的图片（根据缩放策略），然后我们在QLabel上直接绘制矩形。如果图片被缩放了，那么矩形的坐标也会和原始图片的坐标不一致。如果需要获取相对于原始图片的坐标，需要根据QLabel中图片的显示区域进行转换。 2. **图片适应**：如果QLabel设置了缩放（如`setScaledContents(True)`），那么图片会被拉伸。在这种情况下，我们绘制的矩形也会被拉伸。为了避免这种情况，我们通常不设置缩放，而是通过设置`setPixmap`后，让QLabel自动调整大小（或者使用布局管理器）。但是，如果QLabel的大小和图片原始大小不一致，我们需要在绘制矩形时进行坐标转换。 3. **性能**：如果矩形很多，每次重绘都会重绘所有矩形，可能会影响性能。但是通常选框数量不会太多，所以可以接受。 ### 改进：考虑图片缩放 如果QLabel显示图片时进行了缩放，那么鼠标事件的坐标和原始图片坐标不一致。我们需要在绘制矩形时，将矩形的坐标转换到显示区域，或者将鼠标坐标转换到原始图片坐标。这里我们提供一种方法： 在自定义QLabel中，我们可以计算图片在QLabel中的显示区域（保持宽高比）。然后，在鼠标事件中，将鼠标坐标转换为相对于原始图片的坐标。 但是，为了简化，我们假设QLabel显示图片时没有缩放（即图片原始大小），或者将QLabel的大小设置为图片大小。如果必须缩放，那么需要额外的坐标转换。 ### 完整代码示例 以下是一个完整的示例，假设图片以原始大小显示（或QLabel大小与图片一致）： ```python import sys from PyQt5.QtWidgets import (QApplication, QWidget, QVBoxLayout) from PyQt5.QtCore import Qt, QPoint, QRect from PyQt5.QtGui import QPixmap, QPainter, QPen, QColor class ImageLabel(QLabel): def __init__(self, parent=None): super().__init__(parent) self.setMouseTracking(False) self.setAlignment(Qt.AlignCenter) self.origin = QPoint() self.current_rect = None self.rectangles = [] # 存储所有矩形 def setPixmap(self, pixmap): super().setPixmap(pixmap) self.pix = pixmap # 保存原始图片 def mousePressEvent(self, event): if event.button() == Qt.LeftButton: self.origin = event.pos() self.current_rect = QRect(self.origin, self.origin) self.update() elif event.button() == Qt.RightButton: self.rectangles.clear() self.current_rect = None self.update() super().mousePressEvent(event) def mouseMoveEvent(self, event): if self.current_rect is not None and event.buttons() & Qt.LeftButton: self.current_rect = QRect(self.origin, event.pos()).normalized() self.update() super().mouseMoveEvent(event) def mouseReleaseEvent(self, event): if event.button() == Qt.LeftButton and self.current_rect is not None: self.rectangles.append(self.current_rect) self.current_rect = None self.update() super().mouseReleaseEvent(event) def paintEvent(self, event): # 先调用父类绘制图片 super().paintEvent(event) painter = QPainter(self) pen = QPen(QColor(255, 0, 0), 2) painter.setPen(pen) # 绘制所有已完成的矩形 for rect in self.rectangles: painter.drawRect(rect) # 绘制当前正在绘制的矩形 if self.current_rect is not None: painter.drawRect(self.current_rect) class MainWindow(QWidget): def __init__(self): super().__init__() layout = QVBoxLayout() self.label = ImageLabel() layout.addWidget(self.label) self.setLayout(layout) pixmap = QPixmap("image.jpg") # 替换为你的图片路径 self.label.setPixmap(pixmap) # 设置窗口大小（可选） self.resize(pixmap.width(), pixmap.height()) if __name__ == '__main__': app = QApplication(sys.argv) window = MainWindow() window.show() sys.exit(app.exec_()) ``` ### 相关问题 1. 如何在PyQt中实现图片的缩放显示，并保证绘制的矩形与原始图片坐标对应？ 2. 如何将绘制的矩形坐标转换为原始图片上的坐标？ 3. 如何实现除了矩形之外的其他形状（如圆形、多边形）的绘制和保留？ 4. 如何将绘制的多个矩形区域保存为文件或进行进一步处理？</think>### PyQt QLabel 实现鼠标绘制多个选框并保留功能 要实现QLabel上显示图片并支持鼠标绘制多个选框、右键清除所有选框的功能，可以通过以下步骤实现： #### 核心思路 1. 创建自定义QLabel子类，重写鼠标事件和绘制事件 2. 使用列表存储所有选框坐标 3. 实现实时绘制当前选框和保留历史选框 4. 右键点击时清空选框列表 #### 完整代码实现 ```python from PyQt5.QtWidgets import QLabel, QApplication from PyQt5.QtCore import Qt, QPoint, QRect from PyQt5.QtGui import QPainter, QPen, QPixmap, QColor class ImageLabel(QLabel): def __init__(self, parent=None): super().__init__(parent) self.setMouseTracking(True) self.setAlignment(Qt.AlignCenter) # 存储所有选框 self.rectangles = [] # 当前正在绘制的选框 self.current_rect = None self.drawing = False self.origin = QPoint() def setPixmap(self, pixmap): super().setPixmap(pixmap) self.pix = pixmap # 保存原始图片 def mousePressEvent(self, event): if event.button() == Qt.LeftButton: # 开始绘制新选框 self.origin = event.pos() self.current_rect = QRect(self.origin, self.origin) self.drawing = True elif event.button() == Qt.RightButton: # 右键清除所有选框 self.rectangles.clear() self.update() super().mousePressEvent(event) def mouseMoveEvent(self, event): if self.drawing: # 更新当前选框尺寸 self.current_rect = QRect(self.origin, event.pos()).normalized() self.update() # 触发重绘 super().mouseMoveEvent(event) def mouseReleaseEvent(self, event): if event.button() == Qt.LeftButton and self.drawing: # 完成选框绘制，添加到列表 self.rectangles.append(self.current_rect) self.current_rect = None self.drawing = False self.update() super().mouseReleaseEvent(event) def paintEvent(self, event): # 先绘制原始图片 super().paintEvent(event) painter = QPainter(self) pen = QPen(QColor(255, 0, 0), 2) # 红色选框，2像素宽 painter.setPen(pen) # 绘制所有已完成的选框 for rect in self.rectangles: painter.drawRect(rect) # 绘制当前正在绘制的选框 if self.current_rect: painter.drawRect(self.current_rect) ``` #### 使用方法 1. 在主窗口中使用自定义ImageLabel 2. 加载图片并设置到label 3. 实现功能： - 左键拖动：绘制新选框 - 释放左键：保存选框 - 右键点击：清除所有选框 ```python # 在主窗口中的使用示例 from PyQt5.QtWidgets import QMainWindow, QVBoxLayout, QWidget class MainWindow(QMainWindow): def __init__(self): super().__init__() central_widget = QWidget() layout = QVBoxLayout() self.image_label = ImageLabel() layout.addWidget(self.image_label) # 加载图片 pixmap = QPixmap("your_image.jpg") # 替换为实际图片路径 self.image_label.setPixmap(pixmap) central_widget.setLayout(layout) self.setCentralWidget(central_widget) self.setWindowTitle("图片标注工具") self.resize(800, 600) ``` #### 关键点说明 1. **选框存储**：`self.rectangles`列表保存所有已完成选框 2. **实时绘制**：`self.current_rect`跟踪当前正在绘制的选框 3. **右键清除**：在`mousePressEvent`中检测右键点击并清空列表 4. **坐标处理**： - `mousePressEvent`记录起点 - `mouseMoveEvent`更新当前选框 - `mouseReleaseEvent`完成选框 5. **绘制逻辑**： - 先调用`super().paintEvent()`显示图片 - 再绘制所有存储的选框 - 最后绘制当前选框（如果有） #### 注意事项 1. 如果图片被缩放，需要处理坐标转换（示例中假设图片以原始大小显示） 2. 可通过修改`QPen`参数调整选框颜色和粗细 3. 需要确保QLabel设置了正确的图片大小策略 ### 相关问题 1. 如何在PyQt中实现图片缩放功能，并保持选框坐标正确？ 2. 如何将绘制的选框坐标保存到文件或数据库中？ 3. 如何实现选框的编辑功能（移动/调整大小）？ 4. 在PyQt中，如何实现不同颜色的选框表示不同类别？ 5. 如何处理高清大图的显示和标注性能优化？[^1][^2]