Question

我正在尝试编写一个能够显示2D numpy数组图像的预览小部件。该小部件具有固定的大小（正方形），但是图像可以具有任何形状。

它似乎适用于某些图像形状，但对于其他形状则显示无意义，而对于某些其他形状，它会崩溃而没有任何错误消息。

您在我的代码中看到明显的错误吗？

from silx.gui import qt
import numpy


GRAY_COLORTABLE = []
for i in range(256):
    GRAY_COLORTABLE.append(qt.qRgb(i, i, i))


class PreviewImageWidget(qt.QWidget):
    """Preview image"""
    def __init__(self, parent=None):
        super().__init__(parent)
        self.pixmap = qt.QPixmap()
        self.setFixedSize(350, 350)

    def paintEvent(self, event):
        painter = qt.QPainter(self)
        painter.drawPixmap(self.rect(), self.pixmap)

    def setImage(self, img_array):
        # TODO : adjust colortable to actual dtype (autoscale to min - max ??)
        if img_array is None:
            self.pixmap = qt.QPixmap()
        else:
            if img_array.dtype != numpy.uint8:
                max_value = img_array.max()
                img_array = 256. / max_value * img_array
                img_array = img_array.astype(numpy.uint8)

            # binary images are of dtype uint8
            if img_array.max() == 1:
                img_array = img_array * 255
            image = qt.QImage(img_array,
                              img_array.shape[1], img_array.shape[0],
                              qt.QImage.Format_Indexed8)
            image.setColorTable(GRAY_COLORTABLE)
            self.pixmap = qt.QPixmap.fromImage(image)

        self.update()

if __name__ == '__main__':

    app = qt.QApplication([])
    allPreviewWidgets = []

    for sh in [(610, 500), (450, 700), (550, 600),
               (500, 500), (510, 500), (500, 520)]:
        img_array = numpy.zeros(sh, dtype=numpy.uint8)
        img_array[200:350, 250:300] = 1

        previewWidget = PreviewImageWidget()
        previewWidget.setWindowTitle(str(img_array.shape))
        previewWidget.show()
        previewWidget.setImage(img_array)
        allPreviewWidgets.append(previewWidget)

    app.exec_()

几乎方形的形状不起作用。矩形的可以正常工作。在QPainter的文档中说：

注意：如果图像和矩形大小不同。

一个导致程序崩溃的形状示例：（2000，500）

编辑：这是另一个示例，显示了没有QPainter且未调整像素图大小的相同问题。我认为这可以缩小到QImage如何解码numpy数组的问题。

from silx.gui import qt
import numpy

GRAY_COLORTABLE = []
for i in range(256):
    GRAY_COLORTABLE.append(qt.qRgb(i, i, i))


def array2qpixmap(img_array):
    if img_array.max() == 1:
        img_array = img_array * 255
    image = qt.QImage(img_array.astype(numpy.uint8),
                      img_array.shape[1], img_array.shape[0],
                      qt.QImage.Format_Indexed8)
    image.setColorTable(GRAY_COLORTABLE)
    return qt.QPixmap.fromImage(image)


if __name__ == '__main__':

    app = qt.QApplication([])
    labels = []

    for sh in [(610, 500), (450, 700), (550, 600),
               (500, 500), (510, 500), (200, 520)]:
        img_array = numpy.zeros(sh, dtype=numpy.uint8)
        img_array[200:350, 250:300] = 1

        lab = qt.QLabel()
        lab.setFixedSize(700, 700)
        lab.setWindowTitle(str(sh))
        lab.show()
        lab.setPixmap(array2qpixmap(img_array))
        labels.append(lab)

    app.exec_()

Answer 1

在第二种情况下，我只能重现该问题，并且我发现问题出在内存中，因为您在所有转换中都使用相同的对象，因此在某些情况下，内存被消除了，解决方法是要复制数据：

from PySide2 import QtCore, QtGui, QtWidgets
import numpy

GRAY_COLORTABLE = []
for i in range(256):
    GRAY_COLORTABLE.append(QtGui.qRgb(i, i, i))


def array2qpixmap(img_array):
    height, width = img_array.shape
    bytesPerLine, _ = img_array.strides
    image = QtGui.QImage(
        img_array.data.tobytes(),
        width,
        height,
        bytesPerLine,
        QtGui.QImage.Format_Indexed8,
    )
    image.setColorTable(GRAY_COLORTABLE)
    return QtGui.QPixmap.fromImage(image.copy())


if __name__ == "__main__":

    app = QtWidgets.QApplication([])
    labels = []

    for sh in [
        (610, 500),
        (450, 700),
        (550, 600),
        (500, 500),
        (510, 500),
        (200, 520),
    ]:
        img_array = numpy.zeros(sh, dtype=numpy.uint8)
        img_array[200:350, 250:300] = 255
        lab = QtWidgets.QLabel()
        lab.resize(700, 700)
        lab.setWindowTitle(str(sh))
        lab.show()
        lab.setPixmap(array2qpixmap(img_array.copy()))
        labels.append(lab)

    app.exec_()

Answer 2

如果有人遇到与我的第一个示例相同的问题，这是我必须做的才能使其工作。

import numpy
from silx.gui import qt

GRAY_COLORTABLE = [qt.qRgb(i, i, i) for i in range(256)]


class PreviewImageWidget(qt.QLabel):
    """Image preview widget. Displays the image in
    a 2D numpy array with a grayscale colortable.
    """
    def __init__(self, parent=None):
        super().__init__(parent)

        self.size = qt.QSize(150, 150)
        self.setSize(self.size)
        self.pixmap = qt.QPixmap()

    def setSize(self, size):
        self.size = size
        self.setFixedSize(self.size)

    def setImage(self, img_array):
        if img_array is None:
            # null pixmap
            self.pixmap = qt.QPixmap()
        else:
            img_array = img_array.copy()
            bytesPerLine = img_array.strides[0]
            if img_array.dtype != numpy.uint8:
                max_value = img_array.max()
                img_array = 256. / max_value * img_array
                img_array = img_array.astype(numpy.uint8)

            height, width = img_array.shape
            image = qt.QImage(img_array,
                              width, height,
                              bytesPerLine,
                              qt.QImage.Format_Indexed8)
            image.setColorTable(GRAY_COLORTABLE)

            pixmap = qt.QPixmap.fromImage(image)
            self.pixmap = pixmap.scaled(self.size,
                                        qt.Qt.KeepAspectRatio)
        self.setPixmap(self.pixmap)


if __name__ == '__main__':

    app = qt.QApplication([])
    allPreviewWidgets = []

    for sh in [(610, 500), (450, 700), (550, 600),
               (500, 500), (510, 500), (500, 520)]:
        img_array = numpy.zeros(sh, dtype=numpy.uint8)
        img_array[200:350, 250:300] = 255

        previewWidget = PreviewImageWidget()
        previewWidget.setSize(qt.QSize(300, 300))
        previewWidget.setWindowTitle(str(img_array.shape))
        previewWidget.show()
        previewWidget.setImage(img_array)
        allPreviewWidgets.append(previewWidget)

    app.exec_()

在那种情况下，复制数组以确保其在内存中是连续的似乎有所帮助，并且复制QImage似乎是不必要的。

但是在我的真实应用程序中，PyQt5.13失败，原因不明，甚至在设置真实图像之前也是如此。我希望这些都是与当前版本的Qt相关的错误，并且将在下一版本中对其进行修复。

将numpy数组绘制到QWidget

2 个答案: