Как преобразовать изображение PIL в массив Numpy? - PullRequest
Новая
       51

Как преобразовать изображение PIL в массив Numpy?

176 голосов
/ 21 декабря 2008

Хорошо, я играю с преобразованием объекта изображения PIL взад-вперед в массив, чтобы я мог сделать более быстрое преобразование пикселей за пикселем, чем позволял бы объект PixelAccess PIL. Я выяснил, как поместить информацию о пикселях в полезный трехмерный массив с помощью: 

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

Но я не могу понять, как загрузить его обратно в объект PIL после того, как я выполнил все свои потрясающие преобразования. Мне известен метод putdata(), но я не могу заставить его вести себя.

Ответы [ 7 ]

210 голосов
/ 21 декабря 2008

Вы не говорите, как именно putdata() не ведет себя. Я предполагаю, что вы делаете 

>>> pic.putdata(a)
Traceback (most recent call last):
  File "...blablabla.../PIL/Image.py", line 1185, in putdata
    self.im.putdata(data, scale, offset)
SystemError: new style getargs format but argument is not a tuple

Это потому, что putdata ожидает последовательность кортежей, а вы присваиваете ей пустой массив. Это 

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

будет работать, но очень медленно.

Начиная с PIL 1.1.6, «правильный» способ преобразования между изображениями и массивами это просто 

>>> pix = numpy.array(pic)

хотя полученный массив имеет формат, отличный от вашего (в этом случае 3-й массив или row / columns / rgb).

Затем, после внесения изменений в массив, вы сможете выполнить либо pic.putdata(pix), либо создать новое изображение с помощью Image.fromarray(pix).

147 голосов
/ 08 июля 2009

Открыть I в виде массива: 

>>> I = numpy.asarray(PIL.Image.open('test.jpg'))

Сделайте некоторые вещи в I, затем преобразуйте их обратно в изображение: 

>>> im = PIL.Image.fromarray(numpy.uint8(I))

Фильтрация пустых изображений с помощью FFT, Python

Если вы хотите сделать это явно по какой-то причине, есть функции pil2array () и array2pil (), использующие getdata () на этой странице в файле correlation.zip.

46 голосов
/ 17 июня 2017

Я использую Pillow 4.1.1 (преемник PIL) в Python 3.5. Преобразование между Подушкой и NumPy является простым. 

from PIL import Image
import numpy as np
im = Image.open('1.jpg')
im2arr = np.array(im) # im2arr.shape: height x width x channel
arr2im = Image.fromarray(im2arr)

Одна вещь, на которую нужно обратить внимание, это то, что im в стиле подушек является мажорным столбцом, а в стиле numpy im2arr - мажорным строкам. Однако функция Image.fromarray уже учитывает это. То есть arr2im.size == im.size и arr2im.mode == im.mode в приведенном выше примере.

Мы должны позаботиться о формате данных HxWxC при обработке преобразованных числовых массивов, например, выполните преобразование im2arr = np.rollaxis(im2arr, 2, 0) или im2arr = np.transpose(im2arr, (2, 0, 1)) в формат CxHxW.

11 голосов
/ 07 июня 2016

Вам нужно преобразовать ваше изображение в массив numpy следующим образом: 

import numpy
import PIL

img = PIL.Image.open("foo.jpg").convert("L")
imgarr = numpy.array(img)
3 голосов
/ 03 февраля 2017

Пример, который я использовал сегодня: 

import PIL
import numpy
from PIL import Image

def resize_image(numpy_array_image, new_height):
    # convert nympy array image to PIL.Image
    image = Image.fromarray(numpy.uint8(numpy_array_image))
    old_width = float(image.size[0])
    old_height = float(image.size[1])
    ratio = float( new_height / old_height)
    new_width = int(old_width * ratio)
    image = image.resize((new_width, new_height), PIL.Image.ANTIALIAS)
    # convert PIL.Image into nympy array back again
    return array(image)
0 голосов
/ 25 мая 2019 
def imshow(img):
    img = img / 2 + 0.5     # unnormalize
    npimg = img.numpy()
    plt.imshow(np.transpose(npimg, (1, 2, 0)))
    plt.show()

Вы можете превратить изображение в NumPy парсингом изображения в функцию numpy () после выделения функций (ненормализация)

0 голосов
/ 15 февраля 2019

Если ваше изображение хранится в формате Blob (то есть в базе данных), вы можете использовать ту же методику, которая была объяснена Биллом Бегераджем, чтобы преобразовать ваше изображение из Blobs в байтовый массив.

В моем случае мне нужны были мои изображения, хранящиеся в столбце BLOB-объектов в таблице базы данных: 

def select_all_X_values(conn):
    cur = conn.cursor()
    cur.execute("SELECT ImageData from PiecesTable")    
    rows = cur.fetchall()    
    return rows

Затем я создал вспомогательную функцию для изменения набора данных в np.array: 

X_dataset = select_all_X_values(conn)
imagesList = convertToByteIO(np.array(X_dataset))

def convertToByteIO(imagesArray):
    """
    # Converts an array of images into an array of Bytes
    """
    imagesList = []

    for i in range(len(imagesArray)):  
        img = Image.open(BytesIO(imagesArray[i])).convert("RGB")
        imagesList.insert(i, np.array(img))

    return imagesList

После этого я смог использовать byteArrays в моей нейронной сети. 

plt.imshow(imagesList[0])
...