Преобразуйте двумерный массив в 4D - PullRequest
0 голосов
/ 09 ноября 2019

У меня есть матрица y_train (4D), содержимое которой (33, 224, 224, 1) где (n_images, n_pixels_y, n_pixels_x, n_bands), которые необходимо преобразовать в двухмерный вектор с использованием Flatten

Теперь мне нужно сделать обратный процесс и повернуть 2D-вектор обратно в 4D-матрицу с теми же размерами, какой лучший способ это сделать?

Я пытался использовать изменение формы, но это не такочень эффективный

Пример:

y_pred = modelo.predict(x_test)
print(y_pred.shape) #this results in the current shape that is (33, 50176)


img = np.reshape(y_pred[1], (33,224,224,1))
plt.imshow(img, cmap='gray')
plt.show() #this results in a error 'cannot reshape array of size 50176 into shape (33,224,224,1)'

...