Numpy Reshaping меняет изображения

Question

У меня есть массив Numpy, который содержит 2d изображения.Размеры массива numpy: (1000,60,40) (1000 = no.of img).

Я хочу передать этот массив в мою модель, но для этого необходимо, чтобы размеры были (1000,60,40,1) (добавлено 1 для номера каналов).

, поэтому я изменяю массив на

Y=X.reshape(1000,60,40,1)

, так как у меня были неправильные прогнозы. Я проверил , изменив форму массива , чтобы проверить, совпадает ли он с моимorig img, я сделал это, выполнив

Z=Y.reshape(1000,60,40)

И сохранил их в формате PNG, выполнив

for i in range(1000):
misc.imsave('img_rereshaped'+str(i)+'.png',Z[i])

В качестве выходных данных приводятся некоторые файлы png, но они не совпадают с соответствующими оригиналами.из массива X numpy

Неправильно ли я изменяю форму или изменяю ли изменяю входные данные, и повторное изменение измененных данных даст результат, отличный от исходных?

Answer 1

Чтобы проверить, вызывает ли изменение формы проблему, лучше протестировать ее, не вовлекая другие потенциальные ошибки, скажем, от misc.imsave() и т. Д.что переход туда-сюда с использованием изменения формы не вызывает проблем.Может быть, вы не используете PNG сохранить правильно.Возможно, функция ожидает 3 канала, например.Попробуйте построить его локально, используя matplotlib.