Блог

При преобразовании PNG-файлов в массивы numpy я заметил, что преобразование иногда приводило к разным формам массива. В то время как первое изображение имеет трехмерную форму массива (как и большинство изображений), второе имеет два измерения. Оба изображения являются скриншотами и не были изменены. К сожалению, форма массива действительно имеет значение для моего предполагаемого использования.

Я использовал приведенный ниже код для преобразования изображения.

 import numpy as np
import PIL.Image

img_path = ''  # Put the image path here

img_pil = PIL.Image.open(img_path)
img_array = np.asarray(img_pil)

print(img_array.shape)
  

Мой вопрос заключается в том, есть ли способ выровнять формы изображений.

Если ни одно из ваших изображений не является цветным, вы также можете сэкономить память и работать в оттенках серого, чтобы вы могли убедиться, что с:

 im = Image.open(FILENAME).convert('L')
  

Или вы всегда можете перейти к цвету с помощью:

 im = Image.open(FILENAME).convert('RGB')
  

Из того, что я смог увидеть здесь: обработка изображений с помощью numpy, изображения в оттенках серого (как, похоже, в случае со вторым изображением) иногда сохраняются только в двух измерениях, поскольку нет необходимости сохранять три разных цвета для каждого пикселя. Вероятно, это зависит от того, как было создано изображение.