Question

我正在尝试将Python中生成的图像与文件中的图像/照片进行比较。

到目前为止，最好的方法是在Matplotlib中生成一个数字，然后将其转换为numpy数组，并将这些值与我从图像中获得的值进行比较。

我得到以下代码将Matplotlib图形转换为带有RGB通道的3D numpy数组：

def fig2data ( fig ):
    """
    @brief Convert a Matplotlib figure to a 3D numpy array with RGB channels and return it
    @param fig a matplotlib figure
    @return a numpy 3D array of RGB values
    """
    # draw the renderer
    fig.canvas.draw ( )

    # Get the RGBA buffer from the figure
    w,h = fig.canvas.get_width_height()
    buf = numpy.fromstring ( fig.canvas.tostring_rgb(), dtype=numpy.uint8 )
    buf.shape = ( w, h, 3 )

    return buf

其中一个问题 - 到目前为止我想弄清楚的一个问题 - 是这个转换后的图像没有被破坏。例如，如果我绘制一个正方形的所有画布，Matplotlib就会放置这个令人讨厌的框架，并将其转换并混合我的所有结果。

如何只获取我所制作的数字的数值 - 没有任何框架或轴？

或者甚至更好，如果有更简单的方法比较我不知道的NumPy / Matplotlib中的数字和图像，请告诉我。

Answer 1

嗯，这不是使用Matplotlib解决手头问题的真正答案，但是我放弃了这个lib来完成这项工作并且只使用了PIL。

这很容易，尽管它也很慢（但我不知道它是否比Matplotlib慢）。

代码如下：

def makeImage (triangle, largura, altura):
    """
    triangle: receives a tuple in the form: x1, y1, x2, y2, x3, y3, R, G, B, A
    largura: image weight
    altura: image height

    returns: numPy array of the triangle composed final image
    """
    back = Image.new('RGBA', (largura,altura), (0,0,0,0))
    poly = Image.new('RGBA', (largura,altura))
    pdraw = ImageDraw.Draw(poly)

    pdraw.polygon([1,2,3,4,5,6], fill=(255,0,0,127))
    back.paste(poly,mask=poly)

    back = back.convert('RGB')
    backArr = asarray(back)
    #back.show()

    return backArr

如果您知道加快此过程的方法，请告诉我。

将Matplotlib图转换为NumPy数组，不带任何边框/框架或轴

1 个答案: