Как я могу добавить метод выпадения в сетку Keras Retina?

Question

Я работаю с моделью RetinaNet NN для обнаружения объектов и столкнулся с проблемой перебора. Одним из решений является добавление «Dropout». Я использую код keras Здесь

Я хочу добавить Dropout к последним слоям, но я не знаю, как добавить.

Может кто-нибудь помочь, какой файл мне поменять? И как?

Answer 1

Через некоторое время я перепробовал много решений, но ни одно из них не говорило, как именно добавить, поэтому я попытался, а затем нашел, как добавить, поэтому решил ответить на него сам!

Просто нужно добавить такую ​​строку:

outputs = keras.layers.SpatialDropout1D (rate=dropout_rate) (outputs)

Вы можете использовать другой тип выпадения слоя, например: SpatialDropout2D и более.

Answer 2

Вы можете попытаться сохранить полностью связанный слой в переменной, например:

fc1 = model.layers[-3]
fc2 = model.layers[-2]
predictions = model.layers[-1]

Затем создайте свой выпадающий слой и снова соедините их, чтобы построить новую модель, как показано в этом сообщении: Добавьте выпадающие слои между предварительно подготовленными плотными слоями в кератах

Надеюсь, это поможет.