Как заставить getdata() из Pillow возвращать массив из кортежей с RGB цветами пикселя, если фотография чёрно-белая?

Question

[FoxBoost]

Я преобразую массив пискелей, полученный с помощью PIllow в трёхмерный массив NumPy, и данный код нормально работает со цветными изображениями:

def PIL2array(img):
    np.array(img.getdata(), np.uint8).reshape(img.size[0], img.size[1], 3)

Но когда подаётся чёрно-белое изображение, выдаётся следующая ошибка:

ValueError: cannot reshape array of size 784 into shape (28,28,3)

Проблема в том, что Pillow вместо трёх цветов пикселя возвращает один. Все получаемые массивы должны быть одной размерности. Как можно сделать так, чтобы можно было получить три цвета для каждого пикселя в случае чёрно-белого изображения?

Answer 1

[alternativshik]

А если конвертнуть в RGB перед этим?

Comments

[FoxBoost]

А как это можно сделать средствами Pillow?

[alternativshik]

FoxBoost, ответ на этот вопрос находится на первых позициях в гугле, не ленитесь, пожалуйста.

[FoxBoost]

alternativshik, извиняюсь, спасибо за ответ, оставлю итоговый код здесь:

def PIL2array(img):
    if type(img.getdata()[1]) == int:
        img = img.convert('RGB')
    return np.array(img.getdata(),
                    np.uint8).reshape(img.size[0], img.size[1], 3)

[Andrey Dugin]

FoxBoost, используйте нормальный способ:

if img.mode != 'RGB':
    img = img.convert('RGB')

И снова вопрос, зачем преобразовывать картинку в np.array таким кривым способом, если достаточно вызвать np.array(img), без всяких getdata и reshape?

[FoxBoost]

Andrey Dugin, спасибо что поправили, я этот код скопировал с какого-то сайта, ибо reshape не работал на чёрно белых фотографиях, я только потом понял, что проблема именно в том, что они чёрно-белые.

Answer 2

[Andrey Dugin]

Всё гораздо проще:
np.array(img)
Можно ещё поставить -1 вместо 3:

np.array(img.getdata(), np.uint8).reshape(img.size[0], img.size[1], -1)

Comments

[FoxBoost]

Мне нужно было, чтобы массивы всех фотографий были одной размерности (28, 28, 3)

[Andrey Dugin]

FoxBoost, откройте документацию по Pillow, там написано, как перевести ч/б изображение в RGB. Также можете воспользоваться cv2.cvtColor() либо np.repeat().