Question

我想使用PyQT4编写图像标注工具：

从指定文件夹加载大量图像;对于每个图像：
- 用户通过使用鼠标绘制该对象的区域来从图像中选择对象（例如，汽车）
- 选择完成后，对象蒙版将显示在原始图像上
- 当选择完成所有对象时，程序将每个对象蒙版（background：0，foreground：255）保存为单独的png图像
用户应该能够放大/缩小图像

我已经使用wxWidgets在c ++中编写了一个类似的程序（没有放大/缩小）。我对PyQT4＆amp;试图了解事情是如何运作的。即使用户放大/缩小，最困难的部分似乎是绘画和正确获取对象蒙版。

哪个PyQT类适合这个问题？如何正确获取对象掩码（可能是numpy数组）并保存它们？

非常感谢。

按照你的建议，我写了一段代码，用鼠标显示图像并在图像上绘图（仍处于实验和学习阶段）。

我将图像存储在QGraphicsPixmapItem中，将其添加到场景中。然后，我通过覆盖其绘制方法来绘制图像。最后，我覆盖鼠标事件以获取鼠标位置并在那里画一个圆圈。但是当我移动鼠标时，旧圆圈会被删除，并且会绘制一个新圆圈。也就是说，圆圈没有涂在图像上。我认为，我应该使用类似下面的内容，以便绘画在图像上是永久性的：

painter = QPainter()
painter.begin(pixmap)
# here do the drawing
painter.end()

但是，问题是，绘画功能已经把画家作为一个论点;在paint函数中重新创建一个新函数（显然）..

以下是代码：

from PyQt4.QtCore import *
from PyQt4.QtGui import *

class ImageDrawPanel(QGraphicsPixmapItem):
    def __init__(self, pixmap=None, parent=None, scene=None):
        super(ImageDrawPanel, self).__init__()
        self.x, self.y = -1, -1        
        self.radius = 10

        self.pen = QPen(Qt.SolidLine)
        self.pen.setColor(Qt.black)
        self.pen.setWidth(2)

        self.brush = QBrush(Qt.yellow)


    def paint(self, painter, option, widget=None):               
        painter.drawPixmap(0, 0, self.pixmap())                
        painter.setPen(self.pen)
        painter.setBrush(self.brush)        
        if self.x >= 0 and self.y >= 0:
            painter.drawEllipse(self.x-self.radius, self.y-self.radius, 2*self.radius, 2*self.radius)
            self.x, self.y = -1, -1

    def mousePressEvent (self, event):
        print 'mouse pressed'
        self.x=event.pos().x()
        self.y=event.pos().y()            
        self.update()

    def mouseMoveEvent (self, event):
        print 'mouse moving'
        self.x=event.pos().x()
        self.y=event.pos().y()            
        self.update()        

class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()

        self.scene = QGraphicsScene()
        self.scene.setSceneRect(0, 0, 800, 600)

        pixmap=self.openImage()        
        self.imagePanel = ImageDrawPanel(scene = self.scene)
        self.imagePanel.setPixmap(pixmap)
        self.scene.addItem(self.imagePanel)

        self.view = QGraphicsView(self.scene)

        layout = QHBoxLayout()        
        layout.addWidget(self.view)

        self.widget = QWidget()
        self.widget.setLayout(layout)

        self.setCentralWidget(self.widget)
        self.setWindowTitle("Image Draw")

    def openImage(self):
        fname = QFileDialog.getOpenFileName(self, "Open image", ".", "Image Files (*.bmp *.jpg *.png *.xpm)")
        if fname.isEmpty(): return None
        return QPixmap(fname)        

import sys
if __name__ == "__main__":    
    app = QApplication(sys.argv)
    mainWindow = MainWindow()
    mainWindow.show()
    sys.exit(app.exec_())

我现在该怎么办，永久地在图像上画画？我可以存储所有的点并在油漆中重新绘制它们，但这似乎并不高效。我应该在QGraphicsScene中绘图，而不是在QGraphicsPixmapItem本身中绘图吗？

第二个问题是，在绘制图像后，如何获得所选区域蒙版？是这样的，用alpha通道创建一个新图像，然后提取像素值？或者，并行绘制空图像？然后，我还应该跟踪放大/缩小..

Answer 1

我有许多不同的选项，我会从较高级别到较低级别订购：

使用QGraphicsScene，QGraphicsView和QGraphicsItems。这可能是图形密集型操作的主要选项和最佳选择。通过将QGLWidget设置为视口，您甚至可以在许多系统上进行硬件加速。
使用QScrollArea支持放大图片，这可能很简单QLabel。通过更改查看的区域，您将获得有效的缩放。 QLabel可用于绘制图像，但您必须手动跟踪所选区域并执行任何选择叠加。
使用单个QWidget并进行自定义绘画。通过在各种事件后调用update()，您将能够绘制任何必要的更改。期望手动完成几乎所有事情。

我建议使用方法编号1.您可以使用QGraphicsPixmapItem来保存图像。然后，您可以创建表示您的选择的图形项，并使用其边界矩形来查找intersecting区域。 QGraphicsView可以为您处理所有zooming。

PyQT4 - 在图像上绘制区域选择

1 个答案: