Làm cách nào để chuyển đổi một hình ảnh PIL thành một mảng numpy?

Được rồi, tôi đang đùa giỡn với việc chuyển đổi một đối tượng hình ảnh PIL qua lại thành một mảng gọn gàng để tôi có thể thực hiện một số pixel nhanh hơn bằng cách chuyển đổi pixel so với PixelAccessđối tượng của PIL cho phép. Tôi đã tìm ra cách đặt thông tin pixel vào một mảng gọn gàng 3D hữu ích bằng cách:

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

Nhưng dường như tôi không thể tìm ra cách tải nó trở lại vào đối tượng PIL sau khi tôi đã thực hiện tất cả các biến đổi tuyệt vời của mình. Tôi biết putdata()phương pháp này, nhưng dường như không thể khiến nó hành xử.

Bạn không nói chính xác putdata()là không cư xử. Tôi cho rằng bạn đang làm

>>> 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

Điều này là do putdatamong đợi một chuỗi các bộ dữ liệu và bạn đang cung cấp cho nó một mảng gọn gàng. Điều này

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

sẽ làm việc nhưng nó rất chậm.

Kể từ PIL 1.1.6, cách "thích hợp" để chuyển đổi giữa các hình ảnh và mảng numpy chỉ đơn giản là

>>> pix = numpy.array(pic)

mặc dù mảng kết quả có định dạng khác với mảng của bạn (mảng 3-d hoặc hàng / cột / rgb trong trường hợp này).

Sau đó, sau khi bạn thực hiện các thay đổi cho mảng, bạn sẽ có thể thực hiện pic.putdata(pix)hoặc tạo một hình ảnh mới với Image.fromarray(pix).

Mở Idưới dạng một mảng:

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

ISau đó, thực hiện một số nội dung để chuyển đổi nó thành hình ảnh:

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

Lọc các hình ảnh lộn xộn với FFT, Python

Nếu bạn muốn làm điều đó một cách rõ ràng vì một số lý do, có các hàm pil2array () và Array2pil () sử dụng getdata () trên trang này trong tệp tương quan.zip .

Tôi đang sử dụng Gối 4.1.1 (sự kế thừa của PIL) trong Python 3.5. Việc chuyển đổi giữa Gối và numpy rất đơn giản.

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)

Một điều cần chú ý là kiểu Gối imlà cột chính trong khi kiểu numpy im2arrlà hàng chính. Tuy nhiên, chức năng Image.fromarrayđã xem xét điều này. Đó là, arr2im.size == im.sizevà arr2im.mode == im.modetrong ví dụ trên.

Chúng ta nên quan tâm đến định dạng dữ liệu HxWxC khi xử lý các mảng numpy được chuyển đổi, ví dụ: thực hiện chuyển đổi im2arr = np.rollaxis(im2arr, 2, 0)hoặc im2arr = np.transpose(im2arr, (2, 0, 1))thành định dạng CxHxW.

Bạn cần chuyển đổi hình ảnh của mình thành một mảng khó hiểu theo cách này:

import numpy
import PIL

img = PIL.Image.open("foo.jpg").convert("L")
imgarr = numpy.array(img) 

Ví dụ, tôi đã sử dụng ngày hôm nay:

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)

Nếu hình ảnh của bạn được lưu trữ ở định dạng Blob (tức là trong cơ sở dữ liệu), bạn có thể sử dụng kỹ thuật tương tự được Billal Begueradj giải thích để chuyển đổi hình ảnh của bạn từ Blobs sang một mảng byte.

Trong trường hợp của tôi, tôi cần hình ảnh của mình được lưu trữ trong cột blob trong bảng db:

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

Sau đó tôi đã tạo một hàm trợ giúp để thay đổi tập dữ liệu của mình thành 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

Sau này, tôi đã có thể sử dụng byteArrays trong Mạng thần kinh của mình.

plt.imshow(imagesList[0])

Chuyển đổi Numpy to PILhình ảnh vàPIL to Numpy

import numpy as np
from PIL import Image

def pilToNumpy(img):
    return np.array(img)

def NumpyToPil(img):
    return Image.fromarray(img)

def imshow(img):
    img = img / 2 + 0.5     # unnormalize
    npimg = img.numpy()
    plt.imshow(np.transpose(npimg, (1, 2, 0)))
    plt.show()

Bạn có thể chuyển đổi hình ảnh thành numpy bằng cách phân tích hình ảnh thành hàm numpy () sau khi loại bỏ các tính năng (không chuẩn hóa)