如何在mayavi（imshow）中绘制彩色图像

是否可以使用mayavi绘制具有3个颜色通道的图像？根据 mayavi 的文档，mayavi.mlab.imshow只能处理形状为 (n x m) 的图像。

Method

我必须使用 Mayavi 的自定义色彩图，请参阅http://docs.enthought.com/mayavi/mayavi/auto/example_custom_colormap.html http://docs.enthought.com/mayavi/mayavi/auto/example_custom_colormap.html .

我制作了一个颜色图，其中包含原始 (n x m x 3) 图像中的所有像素作为行。我还添加了一个 alpha 通道，以便透明 png 可以正确显示。接下来，我使用灰度图像作为查找表，将像素值作为存储在颜色图中的原始像素的索引。我使用查找表创建了一个 imshow 对象作为输入图像，并用我的自定义颜色图替换了 imshow 对象的颜色图。

Code

对于感兴趣的人来说，这是一些带有测试用例的工作代码。除 pl.imread(...) 处的测试用例外，Pylab 可以在任何地方替换为 Numpy（Pylab 包装 Numpy）。在 Windows 7 上使用 Mayavi 4.3.0 在 Python2.7 上进行了测试。（不幸的是，我必须首先修复 Windows 上的以下 Mayavi 错误https://github.com/enthought/mayavi/pull/96/files https://github.com/enthought/mayavi/pull/96/files).

import pylab as pl
from mayavi import mlab

def mlab_imshowColor(im, alpha=255, **kwargs):
    """
    Plot a color image with mayavi.mlab.imshow.
    im is a ndarray with dim (n, m, 3) and scale (0->255]
    alpha is a single number or a ndarray with dim (n*m) and scale (0->255]
    **kwargs is passed onto mayavi.mlab.imshow(..., **kwargs)
    """
    try:
        alpha[0]
    except:
        alpha = pl.ones(im.shape[0] * im.shape[1]) * alpha
    if len(alpha.shape) != 1:
        alpha = alpha.flatten()

    # The lut is a Nx4 array, with the columns representing RGBA
    # (red, green, blue, alpha) coded with integers going from 0 to 255,
    # we create it by stacking all the pixles (r,g,b,alpha) as rows.
    myLut = pl.c_[im.reshape(-1, 3), alpha]
    myLutLookupArray = pl.arange(im.shape[0] * im.shape[1]).reshape(im.shape[0], im.shape[1])

    #We can display an color image by using mlab.imshow, a lut color list and a lut lookup table.
    theImshow = mlab.imshow(myLutLookupArray, colormap='binary', **kwargs) #temporary colormap
    theImshow.module_manager.scalar_lut_manager.lut.table = myLut
    mlab.draw()

    return theImshow

def test_mlab_imshowColor():
    """
    Test if mlab_imshowColor displays correctly by plotting the wikipedia png example image
    """

    #load a png with a scale 0->1 and four color channels (an extra alpha channel for transparency).
    from urllib import urlopen
    url = 'http://upload.wikimedia.org/wikipedia/commons/4/47/PNG_transparency_demonstration_1.png'
    im = pl.imread(urlopen(url), format='png')
    im *= 255

    mlab_imshowColor(im[:, :, :3], im[:, :, -1])

    mlab.points3d([-200, 300, -200, 300],
                  [-200, 300, 200, -300],
                  [300, 300, 300, 300])
    mlab.show()

if __name__ == "__main__":
    test_mlab_imshowColor()

Results