Получение значений RGBA из .png в Python

Question

Получение значений RGBA из .png в Python

1

Для проекта мне нужно получить значения RGBA из изображения и использовать их в последующих вычислениях. Для этого я использовал im.getdata() и потому, что это дает мне значения в формате списков в списке, я написал код, чтобы поместить его в один длинный список.

for group in pix_val:
    for item in group:
        pix_val_flat.append(item)

Это отлично сработало для меня, пока я не проверил код с совершенно белым изображением. Вместо того, чтобы создавать наборы из 4 значений в списке, он просто дал мне один длинный список - это означает, что я получаю сообщение об ошибке, когда моя программа пытается сгладить список. Кто-нибудь понимает, почему это происходит и как я могу это исправить?

Это мой код:

im=Image.open("/filelocation")
pix_val=list(im.getdata())
pix_val_flat=[]
for group in pix_val:
    for item in group:
        pix_val_flat.append(item)

Это сообщение об ошибке:

Traceback (most recent call last):
    File "/program", line 286 in <module>
        for item in group:
    TypeError: 'int' object is not iterable

Natasha Blakemore 03 дек. 2018, в 18:39

Источник

0

Проблема в группе. Это возвращает int в этом случае. Означает, что pix_val является массивом целых чисел, а не массивом целых чисел. Попробуйте проверить вывод каждого метода с помощью printf. Поймите разницу оттуда.
CSGO 03 дек. 2018, в 17:12
0

Я подозреваю, что ваше чисто белое изображение было преобразовано в изображение палитры, которое имеет только одно значение для каждого пикселя (который является индексом в палитре), а не 4 значения RGBA для каждого пикселя. Если вы хотите изображение RGBA, используйте x=Image.open('name.png').convert('RGBA')
Mark Setchell 03 дек. 2018, в 19:12
0

@MarkSetchell большое спасибо, это многое объясняет. Попробую!
Natasha Blakemore 03 дек. 2018, в 21:43

Показать ещё 1 комментарий

Теги:

python

python-imaging-library

4 ответа

Ещё вопросы

Проблема в группе. Это возвращает int в этом случае. Означает, что pix_val является массивом целых чисел, а не массивом целых чисел. Попробуйте проверить вывод каждого метода с помощью printf. Поймите разницу оттуда.
Я подозреваю, что ваше чисто белое изображение было преобразовано в изображение палитры, которое имеет только одно значение для каждого пикселя (который является индексом в палитре), а не 4 значения RGBA для каждого пикселя. Если вы хотите изображение RGBA, используйте x=Image.open('name.png').convert('RGBA')
@MarkSetchell большое спасибо, это многое объясняет. Попробую!

Liam · Answer 1 · 2018-12-03T16-31-00.000Z

почему бы вам просто не использовать img.load() что упрощает работу с одним img.load().

from PIL import Image

img=Image.open("n.png")
pixels = img.load()

pix_val_flat = [v for c in [[x,y] for x in range(img.size[0]) for y in range(img.size[1])] for v in pixels[c[0],c[1]]]

Natasha Blakemore · Answer 2 · 2018-12-05T09-21-00.000Z

Отредактировал эту строку моего кода, и теперь она работает:

im=Image.open("/filelocation").convert('RGBA')

Отлично сработано! Вы можете принять свой собственный ответ и набрать очки.

Jeremiah · Answer 3 · 2018-12-03T15-56-00.000Z

Попробуйте что-то вроде list(im.getdata()) или numpy.array(im.getdata()) (Вероятно, вам лучше с первым.) Он каким-то образом возвращает один длинный список, а не список списков, и это может смягчить это. Другой вариант - использовать что-то вроде imageio.imread(filename), которое возвращает значения пикселей как массив numpy, а не последовательность. Затем вы можете ссылаться на размеры массива, как вы можете, в любом другом массиве. (например, image[:, :, 0] предоставит вам все ваши значения R).

Benni94 · Answer 4 · 2018-12-03T15-55-00.000Z

Причина, по которой вы получаете сообщение об ошибке, заключается в том, что ваша group переменная в ìnt то момент превращается в значение " ìnt. За это im.getdata() функция im.getdata(), которая суммирует значения RGB с одним значением. Вы можете написать простой обходной путь, чтобы проверить, является ли группа значением int или кортежем (как вы ожидаете). Что-то вроде этого должно работать:

from PIL import Image
im = Image.open("test.png")
pix_val = list(im.getdata())
pix_val_flat = []

for group in pix_val:
    if type(group) == int:
        group (group, group, group, 0) # if int, change it to tupel
    for item in group:
        pix_val_flat.append(item)