2

我正在尝试在 pyqtgraph 的 Dock 中的 ImageView() （或类似的）中显示一个 RGB numpy 数组。

一般的想法是这样的代码：

import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui
import numpy as np

from pyqtgraph.dockarea import Dock, DockArea

class SimDock(Dock):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        #self.im1 = pg.image()
        self.im1 = pg.ImageView()

        self.im1.setImage(np.random.normal(size=(100, 100, 3)))
        self.addWidget(self.im1, row=0, col=0)

        self.im1.ui.histogram.hide()
        self.im1.ui.menuBtn.hide()
        self.im1.ui.roiBtn.hide()


app = QtGui.QApplication([])
win = QtGui.QMainWindow()
area = DockArea()
win.setCentralWidget(area)
win.resize(1500, 800)
win.setWindowTitle('pyqtgraph example: dockarea')

simdock = SimDock("Similar Images", size=(500, 500))
area.addDock(simdock, 'right')

win.show()

# Start Qt event loop unless running in interactive mode or using pyside.
if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        app.instance().exec_()

但是，当我运行上面的那个时，我得到：

ValueError: could not broadcast input array from shape (100,100,3) into shape (100,100)

当我将 RGB 图像切换self.im1为pg.image而不是pg.ImageView在 Dock 中显示时，但我得到了第二个空窗口（我假设它来自pg.image()）。

基于这个问题，ImageView 可以接受 (M, N, 3) RGB 数据，但我似乎无法让它在不弹出第二个窗口的情况下在小部件中显示 RGB 图像。

4

1 回答 1

2

好吧，我发现了一种看起来很合理的方法。 有一篇帖子建议子类pg.ImageView化以便为颜色自动添加查找表。所以，最后我有：

class ColorImageView(pg.ImageView):
    """
    Wrapper around the ImageView to create a color lookup
    table automatically as there seem to be issues with displaying
    color images through pg.ImageView.
    """

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.lut = None

    def updateImage(self, autoHistogramRange=True):
        super().updateImage(autoHistogramRange)
        self.getImageItem().setLookupTable(self.lut)

然后我的代码中的调用变为self.im1 = ColorImageView().

这对我有用，而且看起来相当简单。

于 2018-06-15T16:46:18.610 回答