Tengo una escala de grises de las imágenes del conjunto de datos que yo quería alimentar una red neuronal.
x_train_grey.shape
es (32, 32, 73257)
así que entiendo que es (dimension_x, dimension_y, batch_size)
. Debido a que las imágenes en escala de grises, sólo hay una "profundidad" de la dimensión.
Sin embargo, para alimentar estos datos a la red neuronal debe tener esta forma:(batch_size, dimension_x, dimension_y)
. Con batch_szie
al principio.
¿Cómo puedo cambiar a este formato, por lo que batch_szie
viene antes de la x, y las imágenes de las dimensiones?
Una vez hecho esto, espero ser capaz de pasar esto en una red neuronal (la primera capa es Flatten()
), así:
Flatten(input_shape=(32, 32, 1)),
.
Saludos!