首页 >后端开发 >Python教程 >如何高效地将 PIL 图像转换为 NumPy 数组以进行像素转换?

如何高效地将 PIL 图像转换为 NumPy 数组以进行像素转换?

Mary-Kate Olsen
Mary-Kate Olsen原创
2024-11-12 13:24:02233浏览

How to Efficiently Convert PIL Images to NumPy Arrays for Pixel Transformations?

将 PIL 图像转换为 NumPy 数组以实现高效的像素转换

在处理图像处理任务时,通常需要转换 PIL ( Python 成像库)将图像转换为 NumPy 数组,以实现更快的像素级转换。这种转换可以有效地操作图像数据,从而实现更复杂和时间优化的图像操作。

要将 PIL 图像转换为 NumPy 数组,可以使用以下代码片段:

pic = Image.open("foo.jpg")
pix = numpy.array(pic.getdata()).reshape(pic.size[0], pic.size[1], 3)

此代码从 PIL 图像中读取图像数据,并将其重塑为 3 维 NumPy 数组,其中每个维度代表图像的高度、宽度和通道(例如,RGB)。

要在执行所需的转换后将 NumPy 数组转换回 PIL 图像,可以使用以下代码:

new_pic = Image.fromarray(modified_pix)

但是,此方法不允许使用修改后的 NumPy 数组直接修改原始 PIL 图像。

在 PIL 版本 1.1.6 之前,修改原始 PIL 图像需要将 NumPy 数组转换为元组列表:

data = list(tuple(pixel) for pixel in modified_pix)
pic.putdata(data)

但是,这种方法可能很慢且效率低下。

在 PIL 版本 1.1.6 及更高版本中,在 PIL 图像和 NumPy 数组之间转换的首选方法是:

pix = numpy.array(pic)

此命令生成一个以行、列和 RGB 通道为维度的 3 维 NumPy 数组。

执行像素转换后,可以使用以下方法将更新后的数组转换回 PIL 图像:

new_pic = Image.fromarray(modified_pix)

或者,可以使用修改后的 NumPy 数组修改原始 PIL 图像:

pic.putdata(modified_pix)

以上是如何高效地将 PIL 图像转换为 NumPy 数组以进行像素转换?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn