преобразовать многомерный массив NumPy в изображение

Question

У меня есть пустой массив (trainData.npy) для обработки изображений.Он содержит 2000 изображений, шкалу яркости и высоту 450, ширину 600.

Форма изображения поезда: (2000, 1, 450, 600)

Я ищу способ построить или показать одиниз этих изображений.Я использовал этот код, но я получил TypeError: Неверные размеры для ошибки данных изображения /

import numpy as np
import matplotlib.pyplot as plt

img = img_train[0]
plt.imshow(img)
plt.show()

Answer 1

Просто измените ваше изображение:

import numpy as np
import matplotlib.pyplot as plt

img = img_train[0] #img has dim (ncolor=1, nlines=450, nrows=600)
img = reshape(450, 600) #img has dim (450, 600)
#img = img / img.max #if you need rescaling of greyscale to be in [0..1]
plt.imshow(img)
plt.show()

PS: Лично я нахожу это немного более интуитивным, если изображения упорядочены по размерам (nlines, nrows, ncolorchanel), чем ваш выбор (ncolorchanel, nlines, nrows)