PIL图像到数组（numpy数组到数组）

我有一个.jpg图像，我想转换为Python数组，因为我实现了处理普通Python数组的处理例程 .

似乎PIL图像支持转换为numpy数组，并且根据我写的文档：

from PIL import Image
im = Image.open("D:\Prototype\Bikesgray.jpg")
im.show()

print(list(np.asarray(im)))

这将返回一个numpy数组列表 . 另外，我试过了

list([list(x) for x in np.asarray(im)])

由于它失败了，它什么都没有返回 .

如何从PIL转换为数组，或者简单地从numpy数组转换为Python数组？

4 回答

5
我认为你在寻找的是：
```
list(im.getdata())
```
或者，如果图像太大而无法完全加载到内存中，那么类似的东西：
```
for pixel in iter(im.getdata()):
    print pixel
```
来自PIL documentation：

getdata im.getdata（）=> sequence返回图像的内容，作为包含像素值的序列对象 . 序列对象被展平，因此第一行的值紧跟在第0行的值之后，依此类推 . 请注意，此方法返回的序列对象是内部PIL数据类型，它仅支持某些序列操作，包括迭代和基本序列访问 . 要将其转换为普通序列（例如用于打印），请使用list（im.getdata（）） .
回复于 2024-04-25T03:32:03+08:00

我强烈建议您使用 Image 对象的 tobytes 函数 . 经过一些时间检查后，这会更有效率 .

def jpg_image_to_array(image_path):
  """
  Loads JPEG image into 3D Numpy array of shape 
  (width, height, channels)
  """
  with Image.open(image_path) as image:         
    im_arr = np.fromstring(image.tobytes(), dtype=np.uint8)
    im_arr = im_arr.reshape((image.size[1], image.size[0], 3))                                   
  return im_arr

我在笔记本电脑上播放的时间显示

In [76]: %timeit np.fromstring(im.tobytes(), dtype=np.uint8)
1000 loops, best of 3: 230 µs per loop

In [77]: %timeit np.array(im.getdata(), dtype=np.uint8)
10 loops, best of 3: 114 ms per loop

回复于 2024-04-25T03:32:03+08:00

基于zenpoy's answer：

import Image
    import numpy

    def image2pixelarray(filepath):
        """
        Parameters
        ----------
        filepath : str
            Path to an image file

        Returns
        -------
        list
            A list of lists which make it simple to access the greyscale value by
            im[y][x]
        """
        im = Image.open(filepath).convert('L')
        (width, height) = im.size
        greyscale_map = list(im.getdata())
        greyscale_map = numpy.array(greyscale_map)
        greyscale_map = greyscale_map.reshape((height, width))
        return greyscale_map

回复于 2024-04-25T03:32:03+08:00

我使用numpy.fromiter来反转8灰度位图，但没有副作用的迹象

import Image
import numpy as np

im = Image.load('foo.jpg')
im = im.convert('L')

arr = np.fromiter(iter(im.getdata()), np.uint8)
arr.resize(im.height, im.width)

arr ^= 0xFF  # invert
inverted_im = Image.fromarray(arr, mode='L')
inverted_im.show()

回复于 2024-04-25T03:32:03+08:00

PIL图像到数组（numpy数组到数组） - Python

4 回答

相关问题