Управление Reshape правильно, чтобы построить RNN из предыдущего CNN - PullRequest
Новая
       37

Управление Reshape правильно, чтобы построить RNN из предыдущего CNN

1 голос
/ 02 июля 2019

В настоящее время я пытаюсь построить CRNN с помощью Keras.Когда я пытался изменить размер входного файла, у меня возникли проблемы с поиском правильного размера для моего LSTM.После некоторой отладки я обнаружил поле в моем модель объекте с именем output_shape , значение которого было (3,1244), и я попытался передать его как двумерный массив с (3224).Все работало нормально, но я правильно сделал?Что за математика стоит за этим и что я могу сделать в следующий раз, чтобы обнаружить этот размер без отладки? 

def CRNN(blockSize, blockCount, inputShape, trainGen, testGen, epochs):

model = Sequential()

# Conv Layer
channels = 32
for i in range(blockCount):
    for j in range(blockSize):
        if (i, j) == (0, 0):
            conv = Conv2D(channels, kernel_size=(5, 5),
                          input_shape=inputShape, padding='same')
        else:
            conv = Conv2D(channels, kernel_size=(5, 5), padding='same')
        model.add(conv)
        model.add(BatchNormalization())
        model.add(Activation('relu'))
        model.add(Dropout(0.15))
        if j == blockSize - 2:
            channels += 32
    model.add(MaxPooling2D(pool_size=(2, 2), padding='same'))
    model.add(Dropout(0.15))

# Feature aggregation across time
model.add(Reshape((3, 224)))

# LSTM layer
model.add(Bidirectional(LSTM(200), merge_mode='ave'))
model.add(Dropout(0.5))

# Linear classifier
model.add(Dense(4, activation='softmax'))


model.compile(loss=keras.losses.categorical_crossentropy,
              optimizer=keras.optimizers.Adam(),
              metrics=['accuracy']) # F1?


model.fit_generator(trainGen,
                    validation_data=testGen, steps_per_epoch = trainGen.x.size // 20,
                    validation_steps = testGen.x.size // 20,
                    epochs=epochs, verbose=1)
return model

# Function call
model = CRNN(4, 6, (140, 33, 1), trainGen, testGen, 1)
...