Próbuję ustawić poprawny kształt wejściowy na pierwszą warstwę LSTM w Kerach, ale jest to trudne dla mnie, aby zrozumieć, jakie jest prawidłowe wejście_sztywne dla druku (X_Train.Shape), które dostaję (9600, 64, 64, 1). .

0
Michale Osmanoglu 15 marzec 2021, 17:50

1 odpowiedź

Najlepsza odpowiedź

Dane te nie mogą przejść do warstwy LSTM, które oczekuje danych 3D. Może spróbuj tf.keras.layers.ConvLSTM2D po dodaniu kroków czasowych wymiar:

import tensorflow as tf

images = tf.random.uniform((10, 1, 224, 224, 1))

classifier = tf.keras.Sequential([
    tf.keras.layers.ConvLSTM2D(8, 
                               kernel_size=(3, 3),
                               input_shape=(1, 224, 224, 1), 
                               return_sequences=True)
])

classifier(images)
   [[[[-3.53521258e-02, -2.02189311e-02, -2.47801729e-02, ...,
           -2.34759413e-03,  4.60262299e-02,  4.76470888e-02],
          [ 1.04620471e-03, -9.23185516e-03,  1.37878451e-02, ...,
           -4.88127321e-02,  4.20494527e-02,  6.06664363e-03],
          [ 1.26057174e-02,  1.07498122e-02, -1.85700115e-02, ...,
           -1.49483923e-02,  1.21065099e-02,  1.71790868e-02]...,
 
0
Nicolas Gervais 15 marzec 2021, 15:21